/FocalLoss-tf-pytorch

A PyTorch and Tensorflow implementation of focal loss

Primary LanguagePython

FocalLoss-tf-pytorch

A PyTorch and Tensorflow implementation of focal loss by Lin et. al. in their Focal Loss for Dense Object Detection paper.

Requirements

tensorflow 2.0

pytorch