Metadata-Version: 2.1
Name: unified-focal-loss-pytorch
Version: 0.1.0
Summary: 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
Author: Taylor Denouden
Author-email: taylordenouden@gmail.com
Requires-Python: >=3.9,<4.0
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: einops (>=0.6.1,<0.7.0)
Requires-Dist: torch (>=2.0.1,<3.0.0)
Description-Content-Type: text/markdown

# Unified Focal Loss PyTorch

*WORK IN PROGRESS - NOT PROPERLY TESTED*

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

Extended for multiclass classification and to allow passing an ignore index.

[1]: https://github.com/mlyg/unified-focal-loss

