U-Net architecture fail to segment the test images.

4 visualizaciones (últimos 30 días)
Raza Ali
Raza Ali el 25 de Jun. de 2021
Comentada: Raza Ali el 1 de Jul. de 2021
I am trainng U-Net architecture for semantic segmentation but after training the trained network does not segment the test images.
I am using 500 images (400 training, 50 validation and 50 testing) along with labelled mask to train the network. The training process reaches at 90% validation accuracy after 2nd epoch and remains at same level for rest of the epochs. I tried many possible ways including change in learning rate but it does not work. The images are in JPEG format and labeld mask has png format.
load gTruth % load gTruth file of training data
[imdsTrain,pxdsTrain] = pixelLabelTrainingData(gTruth);
pximdsTrain = pixelLabelImageDatastore(imdsTrain,pxdsTrain);
load gTruth % load gTruth file of Validiation data
[imdsVal,pxdsVal] = pixelLabelTrainingData(gTruth);
pximdsVal = pixelLabelImageDatastore(imdsVal,pxdsVal);
load gTruth % load gTruth file of test data
[imdsTest,pxdsTest] = pixelLabelTrainingData(gTruth)
options = trainingOptions('sgdm', ...
'LearnRateSchedule','piecewise',...
'LearnRateDropPeriod',2,...
'LearnRateDropFactor',0.1,...
'Momentum',0.9, ...
'InitialLearnRate',1e-3, ...
'L2Regularization',0.09, ...
'ValidationData',pximdsVal,...
'MaxEpochs',5, ...
'MiniBatchSize',2, ...
'Shuffle','every-epoch', ...
'VerboseFrequency',50,...
'Plots','training-progress',...
'ValidationFrequency',100);
imageSize = [256 256 1];
numClasses = 2;
Network = unetLayers(imageSize,numClasses)
I repalced cross entropy loss with dice los function.
rng(1) %
Trained_net= trainNetwork(pximdsTrain,Network,options);

Respuesta aceptada

ytzhak goussha
ytzhak goussha el 30 de Jun. de 2021
This looks like a case of data imballance.
Check you data, see if there is a dominant class such as back ground you can also look at the predictions and see if they are all the same label.
If there is imballance, you need to use weight in your loss functions compensate for that, and there are other ways
See:
  3 comentarios
ytzhak goussha
ytzhak goussha el 1 de Jul. de 2021
I'm sorry I didn't understand. Why not use weighted loss? the weight don't have to be exact, so you don't have to recalculate them with every sample.
Raza Ali
Raza Ali el 1 de Jul. de 2021
I have used Balanced cross entropy loss and to comapre the performacne of other loss fucntion I need to implemnt Dice and cross entropy loss as well.

Iniciar sesión para comentar.

Más respuestas (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by