search

unsky / focal-loss

Focal loss for Dense Object Detection
483 stars 124 forks source link

How to use focal loss in Caffe Models? #10

Open RichardMrLu opened 6 years ago

RichardMrLu commented 6 years ago

Just like YOLO or SSD, how to replace original loss function with focal loss in Caffe? Tanks for your answer,this question puzzle me ..

ViewSky commented 6 years ago

You can rewrite FOCAL LOSS layer and use it with prototxt. In SSD, you need to rewrite util.cpp with new loss function

RichardMrLu commented 6 years ago

Thanks for your answer.