Home

Awesome

Focal Loss

This is the keras implementation of focal loss proposed by Lin et. al. in their Focal Loss for Dense Object Detection paper.

Usage

You have to compile your model with focal loss. Sample:

model_prn.compile(optimizer=optimizer, loss=[focal_loss(alpha=.25, gamma=2)])