Hsuxu / Loss_ToolBox-PyTorch

PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss

Loss_ToolBox

Introduction

This repository include several losses for 3D image segmentation.

1. Focal Loss (PS:Borrow some code from c0nn3r/RetinaNet)
2. Lovasz-Softmax Loss(Modify from orinial implementation LovaszSoftmax)
3. DiceLoss