search

bonlime / keras-deeplab-v3-plus

Keras implementation of Deeplab v3+ with pretrained weights
MIT License
1.35k stars 429 forks source link

Argmax giving the same result on each pixel #116

Open TimbusCalin opened 4 years ago

TimbusCalin commented 4 years ago

When applying argmax on logits, the same result is for every pixel in the image.

Does anyone know how to solve this?

TimbusCalin commented 4 years ago

I also mention that the accuracy is very high during the training and the masks are prepared accordingly(tested with other neural network).

graffam commented 4 years ago

how many classes are you using, are you applying softmax or sigmoid to those logits?

TimbusCalin commented 4 years ago

30 classes and applying softmax not sigmoid; to the best of my knowledge sigmoid can be applied in case of overlapping classes, just like in multilabel classification.

TimbusCalin commented 4 years ago

Will test again with new testing configuration.