• Stars
    star
    331
  • Rank 127,323 (Top 3 %)
  • Language
    Python
  • License
    MIT License
  • Created about 7 years ago
  • Updated over 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Focal Loss implementation in Keras

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)])