Only getting black image predictions when n_classes = 2

[bc-bytes]

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.