Yes, thank you. Finally, I got it.
Can you tell me how can I improve performance of my Network. These are the current results that I a getting.

  
    
    
    UNet Segmentation Results not good vision
  
  
    Hi, I am trying to work on Segmentation with Unet on CamVid Dataset. Data has 32 classes, masks available are of [n,3,h,w]. I converted the masks in [n,32,h,w]. While during training using CrossEntropyloss Ii game target to criterion as target=target.argmax(1) which makes target of shape [n,1,h,w] while my model predictions are of shape [n,32,h,w]. But when test on test data result is not good. 
 [download (1)] 
[download (2)] 
My code is Given: 
def Double_Conv(input_channel, output_channel):
r…