Save finalized training progress plots using OutputFcn

Question

Teo 2021-5-16

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/831163-save-finalized-training-progress-plots-using-outputfcn

评论： Michael Alvarez 2022-10-6

Hi,

i currently using the OutputFcn to save the finalised training progress as a figure, however i noticed that the saved figure are not the finalised result of the training progress instead it is the figure before reaching the finalised training with the validation accauracy. The result that i obtained is as followed.

Obtained figure:

Desired Figure (with validation accuracy):

Attached is also the code that i'm using in the live editor.

options = trainingOptions('sgdm', ...
    'MiniBatchSize',miniBatchSize, ...
    'MaxEpochs',2, ...
    'InitialLearnRate',1e-4, ...
    'Shuffle','every-epoch', ...
    'ValidationData',augimdsValidation, ...
    'ValidationFrequency',valFrequency, ...
    'Verbose',false, ...
    'Plots','training-progress', ...
    'OutputFcn',@(info)SaveTrainingPlot(info));
% ... Training code ...
% At the end of the script
function stop = SaveTrainingPlot(info)
stop = false;
if info.State == "done"
    currentfig = findall(groot,'Type','Figure');
    savefig(currentfig,'test.fig')
end
end

The problem seem to be related to info.State == "Done", but is there any workaround to save the desired output as figure/image?

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

请先登录，再进行评论。

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

Answer 1

Srivardhan Gadila 2021-5-23

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/831163-save-finalized-training-progress-plots-using-outputfcn#answer_706518

在 MATLAB Online 中打开

I think that the final training plot is updated after exiting the function mentioned for the 'OutputFcn' argument. You can check it yourself by inserting pause(10) in the function and observe that the trianing progress plot updation will also pause for 10 seconds.

As you are interested in the final training progress plot, instead of placing the commands findall and savefig inside 'OutputFcn' function just place them after the trainNetwork function.

[net,info] = trainNetwork(___)
currentfig = findall(groot, 'Tag', 'NNET_CNN_TRAININGPLOT_UIFIGURE');
savefig(currentfig,'test.fig');

4 个评论
显示 2更早的评论隐藏 2更早的评论

Michael Alvarez 2022-10-6

在 MATLAB Online 中打开

I have the same problem with the "fig" image, but thinking of a solution, I decided to copy the screen and save it as a "png" image. Maybe this helps other people.

currentfig = findall(groot, 'Tag', 'NNET_CNN_TRAININGPLOT_UIFIGURE');
img = screencapture(0, 'Position', currentfig(1,1).Position);
imwrite(img,"TrainingProgress.png");