unified-focal-loss-pytorch

An implementation of loss functions from "Unified Focal loss: Generalising Dice and cross entropy-based losses to handle class imbalanced medical image segmentation"


License
MIT
Install
pip install unified-focal-loss-pytorch==0.1.2