How can I test the network I've trained?

Question

Ferhat Kangal 2022-12-29

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1886042-how-can-i-test-the-network-i-ve-trained

回答： Christopher McCausland 2022-12-29

Hello everybody, I have trained a CNN model which is Resnet50 and now I want to test it with different dataset and get accuracy. Anyone has any idea how can I do that? I have created a imageDataStore called imdsTest which has 5 classes and thousands of images in the classes.

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

请先登录，再进行评论。

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

Answer 1

Christopher McCausland 2022-12-29

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1886042-how-can-i-test-the-network-i-ve-trained#answer_1138092

在 MATLAB Online 中打开

Hi Ferhat,

If you have trained the model it will save to your workspace as a fancy variable 'model'. You can use this to then test with images it hasn't seen before.

% Lets generate a network -- Yours may look slightly diffrent, thats okay
miniBatchSize  = 128;
validationFrequency = floor(numel(YTrain)/miniBatchSize);
options = trainingOptions('sgdm', ...
    'MiniBatchSize',miniBatchSize, ...
    'MaxEpochs',30, ...
    'InitialLearnRate',1e-3, ...
    'LearnRateSchedule','piecewise', ...
    'LearnRateDropFactor',0.1, ...
    'LearnRateDropPeriod',20, ...
    'Shuffle','every-epoch', ...
    'ValidationData',{XValidation,YValidation}, ...
    'ValidationFrequency',validationFrequency, ...
    'Plots','training-progress', ...
    'Verbose',false);
% Lets train it;
net = trainNetwork(XTrain,YTrain,layers,options);
% Finally lets test our network;
YPredicted = predict(net,imdsTest);