Only getting black image predictions when n_classes = 2

IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.

MIT License

86 stars 19 forks source link

Only getting black image predictions when n_classes = 2 #23

Open bc-bytes opened 2 years ago

bc-bytes commented 2 years ago

I tested with several model configurations, with and without pretrained weights, but am only getting black images for prediction outputs. Is this library not suitable for binary classification (background = 0, object = 1)? Any help much appreciated.