Image Regression

Hello everyone!

I have a couple of beginner questions.

  1. I have a csv file containing: 1st column: file names of images; 2nd column values 0 - 100. I want to create a DataBunch to do an image regression. Is the following code correct for creating a suitable DataBunch?

data = (ImageList.from_csv(path, ‘files_and_values.csv’)
.split_by_rand_pct(valid_pct=0.2, seed = 42)
.label_from_df(label_cls = FloatList)
.transform(tfms=[], size=224)
data = data.normalize(imagenet_stats);

  1. How to create a DataBunch for doing image regression with the csv file containing additional 3rd and 4th column of numbers.

  2. Is it possible to use y_range = [50,102] with image regression, likewise performed in learn = collab_learner(data, n_factors=50, y_range=y_range)?

Thank you!

1 Like