/focal-loss.pytorch

A PyTorch Implementation of Focal Loss.

Primary LanguagePythonMIT LicenseMIT

Focal Loss for Dense Object Detection in PyTorch

This repository is forked from here. It is slightly modified so that it can be run using PyTorch-1.0 and Python-3.

-focal loss