UNet training

[sctrueew]

Hi,

I trained a model with the public dataset but the result is strange. Could you please some tips for training.

Thanks.

I have one class and I set the classes param to 2

unet_model = Unet(encoder_name="resnet50", encoder_weights="imagenet", decoder_channels=(256, 128, 64, 32, 16),
                  in_channels=3, classes=2)

--width: 512
--height: 512
--epoch: 30
--batchsize: 2

dataset sample: 900 images

Result: