This repository include several losses for 3D image segmentation.
- Focal Loss (PS:Borrow some code from c0nn3r/RetinaNet)
- Lovasz-Softmax Loss(Modify from orinial implementation LovaszSoftmax)
- DiceLoss (To Be Released)
xinyu-ch/Loss_ToolBox-PyTorch
PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss
Jupyter NotebookMIT