Binary segmentation may only output zeros

[wptoux]

Describe the bug
The EncoderDecoder Segmentor seems not to take into account the binary segmentation case. The argmax linked below may only return 0 when num_classes=1 and using sigmoid as loss fuction.

mmsegmentation/mmseg/models/segmentors/encoder_decoder.py

Line 263 in 0e37281

seg_pred = seg_logit.argmax(dim=1)

mmsegmentation/mmseg/models/segmentors/encoder_decoder.py

Line 286 in 0e37281

seg_pred = seg_logit.argmax(dim=1)

[MeowZheng]

Thanks for your bug reporting. We will fix it asap.

@xiexinch please have a look.

[xiexinch]

Hi @wptoux,
We are working on fixing this bug, but it takes some time.

[GuardSkill]

Me too, we alter code to make a multi-label binary segmentation, but the per pixel in binary preds has the same values.

[xiexinch]

Hi @wptoux, @GuardSkill,
We create PR #2016 to fix the bug, could you try it and feedback to us?