Prediction differs during training with result

2 次查看（过去 30 天）

显示更早的评论

Christian Huggler 2022-5-16

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1720410-prediction-differs-during-training-with-result

评论： Abhijit Bhattacharjee 2022-5-19

在 MATLAB Online 中打开

I have twelve weighted classes that I train with a large augmented training and validation pixelLabelImageDatastore.

Created with:

lgraph=deeplabv3plusLayers(imageSize, numel(classes), 'resnet18');
lgraph = replaceLayer(lgraph, "classification", pixelClassificationLayer('Name','labels','Classes',tbl.Name,'ClassWeights',classWeights));
lgraph = replaceLayer(lgraph, "data", imageInputLayer(imageSize,"Name","data","Normalization","none"));

The training accuracy converges very fine to about 99,3% (98,5% - 99,7) and the loss to about 0.05 (for both training and validation).

When I test the generated DAGNetwork with "jaccard", only the first ten classes have high IOU, and the last 2 are zero! I also tested different normalizations such as zscore - always the same result. When I use the "predict" or "semanticseg" functions to check individual images, classes 11 and 12 seem to be poorly learned indeed.

But if I set a breakpoint in the "forwardLoss" function in "SpatialCrossEntropy.m" during the training and examine e.g. class 11 with "imshow(Y(:,:,11))", everything is fine learned!

What happens in "trainNetwork()" when the training is finished? Under what circumstances do forwardLoss() scores differ?