Prediction Mask not getting displayed in Unet- Please Help

[ramanathan anand]

Hello All,

I am trying to implement Unet based segmentation for Medical Histopathology classification for single class problem. I get the loss and accuracy of the model but predicted masks are not getting displayed at all. I am using Batch Size =32, Adam optimizer with 0.0001 learning rate and Sigmoid activation function. Please help.