/pytorch-multi-class-focal-loss

An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.