I am trying road segmentation. I have trained a UNet like architecture. The loss becomes very less in two epochs itself. But when I try it on some data the output is not as expected. Only some dotted lines appear whereas the output should be a black and white image.