/pytorch-multi-class-focal-loss

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

Primary LanguagePythonMIT LicenseMIT

Watchers