I forgot to mention one more thing. While obtaining my label I transform it to tensor which divides each of the pixel value in it by 255.  This changes my label values from [0 1] to [0 , 1/255]. I manually change this back by assigning label[label!=0]=1, followed by label=label.long(). This label is then passed into the loss function. Dont know if this would change anything.
After much search I came across another post with the same problem , but there doesn’t seem to be a solution posted.

  
      github.com/tensorflow/models
  
  
    



Issue: deeplab doesn't predict correctly the segmentation masks 20


	
	opened by kirk86
	on 2018-03-25
	
	
	


Everything seems to be working properly training/evaluation etc. except from the fact that deeplab doesn't predict the segmentation masks.
Example:

The original images...