How to prepare imagedatastore for Image to single value regression

Question

qi lu 2020-8-5

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/575611-how-to-prepare-imagedatastore-for-image-to-single-value-regression

编辑： David 2022-5-11

I would like to train a CNN for image regression using a datastore. The output is a single value. I set the Labels in imagedatastore to a value vector(Y_train). But the debug information show the labels in imagedatastore must be a categorical vetcor. Is it possible to use the imagedatastore for regression problems.

ims_train = imageDatastore('F:\imagestore\train_image224', 'FileExtensions', '.tiff', 'Labels', Y_train, 'ReadSize', 128);

% here I defined my network architecture

% here I defined my training options

[net,Info] = trainNetwork(ims_train,lgraph,options);

3 个评论
显示 1更早的评论隐藏 1更早的评论

Cris LaPierre 2021-1-5

编辑：Cris LaPierre 2021-1-5

Question from Sho Wright

Hi Qi Lu, I'm also trying to use imds and augimds to preprocess images for an image-to-value regression network, could you share your code if possible?

Cris LaPierre 2021-1-5

Qi lu's reply

Sorry. I don't find any solutions or workaround. It seems it's impossible to use imds to solve regression problems since the he labels in imagedatastore must be a categorical vetcor. But the labels in regression problems are values. Now I just load all image in the RAM and process image.

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Joe Chalfoun 2021-8-12

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/575611-how-to-prepare-imagedatastore-for-image-to-single-value-regression#answer_765952

编辑：Joe Chalfoun 2021-8-12

This solution worked for me:

Just get the list of files name from the regular imagedatastore and create a table of two columns, the first one is the files name and the second column is the regression response. You can then input that table to trainNetwork function or for the validation option too, like this:

For training:

[net, info] = trainNetwork(table(ims_train.Files,Y_train),lgraph,training_options);

to add validation data:

options = trainingOptions('ValidationData',table(ims_val.Files,Y_val));

Hope this helps.