Loss_ToolBox-PyTorch
Loss_ToolBox-PyTorch copied to clipboard
PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss
Loss_ToolBox
Introduction
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