Binary Image Semantic Segmentation

[ghost]

I have a problem of image segmentation in which I have only one class means its mask and other is background. I have changed the number of classes in model and to 2 but it show miou=1 and result are worst.

[bonbonjour]

I want to train my own data with 16 class,where should be changed in the code ? Thanks

[margokhokhlova]

@Murad1997 Did you solve this? I am curently working on the same issue, I think, the softmax somewhere should be replaced by a sigmoid for a binary classification problem, but I am not sure

[LbinB]

did you solve the problem? i am face the same issue,could you help me if you already know how to do @rishizek @margokhokhlova @bonbonjour @Murad1997

[margokhokhlova]

@LbinB I used this code for binary segmentation, fine-tuning the thing, but I didn't get very good results (a bit less accurate than a simple unet). The number of classes even for a binary segmentation was 2, and I just the replaced last activation layer.