mkocabas / focal-loss-keras

Focal Loss implementation in Keras

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Focal Loss implementation in Keras

License:MIT License


Languages

Language:Python 100.0%