Skip to content

Latest commit

 

History

History
9 lines (8 loc) · 378 Bytes

README.md

File metadata and controls

9 lines (8 loc) · 378 Bytes

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=2, gamma=2)])