top of page

Loss Function

I mixed two Loss function for the segmentation. dice loss and CrossEntropyLoss.

​

Dice loss is compared to the distance of two tensors ( predict and true mask).

Screenshot 2020-08-02 at 9.30.24 PM.png

ref :usuyama. (2018). pytorch-unet. Github. https://github.com/usuyama/pytorch-unet

bottom of page