search

zacario-li / Segmentation-based-deep-learning-approach-for-surface-defect-detection

Segmentation-based deep-learning approach for surface-defect detection with pytorch
MIT License
22 stars 2 forks source link

cross-entropy for segmentation is failed #2

Closed zacario-li closed 4 years ago

zacario-li commented 4 years ago

Use cross-entropy for segmentation network is failed, I'll fix it later

zacario-li commented 4 years ago

fixed, if you want to use ce for loss function, clamp your labels into (0,1) before put them into criterion.