I design a net for semantic segmentation by myself,and train on pascal voc 2012.
I crop the image (512 512),batch_size=1. I
use BatchNorm2d to promte convergence.
I use the loss function is  NLLLoss2d( log.softmax (output), target).
I use SGD ,lr=0.001,momentum=0.9.
After 17 epochs the loss value very small .Then I got a good model on train set,but I got worst result when I use the model on val set,and I try the model of different loss value on val set .
I think the problem is not overfitting.
Is ther any body give me some suggestions
This link is my picture show

  
      
      GitHub
  
  
    

mshmoon/The-show-of-my-semantic-segmentiaon 11

Contribute to The-show-of-my-semantic-segmentiaon development by creating an account on GitHub.