Skip to content

Focal Loss of multi-classification in tensorflow

Notifications You must be signed in to change notification settings

zhezh/focalloss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Focal Loss

Lin, T.-Y., Goyal, P., Girshick, R., He, K., & Dollár, P. (2017). Focal Loss for Dense Object Detection, 130(4), 485–491. https://doi.org/10.1016/j.ajodo.2005.02.022

Implementation for focal loss in tensorflow.

This focal loss is a little different from the original one described in paper. This one is for multi-class classification tasks other than binary classifications.

The input are softmax-ed probabilities.

Releases

No releases published

Packages

No packages published

Languages