search

yassouali / CCT

:page_facing_up: Semi-Supervised Semantic Segmentation with Cross-Consistency Training (CVPR 2020).
https://yassouali.github.io/cct_page/
MIT License
395 stars 58 forks source link

ohem loss #35

Closed gbyy422990 closed 3 years ago

gbyy422990 commented 3 years ago

hi, Its really a nice work.

CCT does NOT work when I changed supervised loss function from cross-entropy to OHEM, the mean-IoU on eval dataset seems like really close between full-sup and semi-sup, I dont know why this happen? And by the way, have u ever tried ohem loss?

yassouali commented 3 years ago

Hi @gbyy422990 unfortunately, I did not test the method with ohem loss, so I can't say for sure why that is the case. But maybe you'll need to modify the hyperparameters a little bit to find better results. Or maybe implement a CCT loss with a hard example mining instead of computing it over the whole unlabeled batch.

Good luck.