What is the correct syntax for using an augmentedImageDatastore as validation data in trainingOptions for the trainNetwork() function?

Question

Dominic Maguire 2020-5-12

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/524951-what-is-the-correct-syntax-for-using-an-augmentedimagedatastore-as-validation-data-in-trainingoption

评论： Dominic Maguire 2020-5-15

I have the following image datastore:

myDataset = imageDatastore('C:\MyData',...
    'IncludeSubfolders',true,...
    'FileExtensions','.png',...
    'LabelSource','foldernames');
[imdsTrain, imdsVal, imdsTest] = splitEachLabel(myDataset, 0.6, 0.2, 'randomized');

Which I then prepare to input into some pre-trained networks:

augimdsTrain = augmentedImageDatastore([227 227 3],imdsTrain,...
    'ColorPreprocessing', 'gray2rgb',...
    'DataAugmentation', augmenter);
augimdsVal = augmentedImageDatastore([227 227 3],imdsVal, ...
    'ColorPreprocessing', 'gray2rgb');
augimdsTest = augmentedImageDatastore([227 227 3],imdsTest,...
    'ColorPreprocessing', 'gray2rgb');

What is the correct syntax for using augimdsVal as 'ValidationData'in trainingOptions? Do I just use the augmented datastore as is (as I've seen in some of the documentation):

options = trainingOptions('sgdm','InitialLearnRate', 0.0001, 'ValidationData', augimdsVal);

Or can I add labels like so:

options = trainingOptions('sgdm','InitialLearnRate', 0.0001, 'ValidationData', {augimdsVal imdsVal.Labels});

Or is there a better way?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Harsha Priya Daggubati 2020-5-15

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/524951-what-is-the-correct-syntax-for-using-an-augmentedimagedatastore-as-validation-data-in-trainingoption#answer_432606

Hi,

The name-value pair 'ValidationData' can support only the following types as a value:

Image Data Store.
Data Store.
Table.
Cell array of X and Y, where X represents data while Y represents corresponding labels.

So, I guess it is invalid to give labels again as a value.

Refer to the following link:

https://www.mathworks.com/help/deeplearning/ref/trainingoptions.html#namevaluepairarguments

Hope this clarifies your concern!

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

Harsha Priya Daggubati 2020-5-15

As the imagedataStore has labels, there is no need to explicitly specify labels to augmented image datastore, when we use imds as the argument during creation.

Dominic Maguire 2020-5-15