zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

change the loss function #150

Open Easonrust opened 5 years ago

Easonrust commented 5 years ago

when the target area of ​​the segmentation is particularly small on the whole picture, resulting in a serious imbalance of samples, so we change the loss function to ‘dice coef loss’