VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Background class is predicted as one of foreground classes

DISAPPEARED13 opened this issue · comments

Hi, there, I use Deeplabv3+ as a multiple class semantic segmentation, and background class is regarded as label 0, while the other is from 1 to 13.
But in validation, the background class is predicted as class 4. It's really confusing.

some training details:
SGD optimizer
learning rate is 0.001
data preprocess issue is refered to nnunet.
loss: only Dice loss can work, while others were collapsed.

Thanks! Really thanks a lot!

I fixed this problem. sorry to bother. Thanks a lot!