OHEM and focal loss, which is better?

cedrickchee / ssd-yolo-retinanet

Multi-class object detection pipeline—Single Shot MultiBox Detector (SSD) + YOLOv3 (real-time) + focal loss (RetinaNet) + Pascal VOC 2007 dataset

18 stars 8 forks source link

OHEM and focal loss, which is better? #2

Open chengyu0910 opened 5 years ago

chengyu0910 commented 5 years ago

Hi, is the focal loss works in SSD? How much improvement is reached in VOC2007 evaluation when SSD is trained with focal loss? I want to know whether the focal loss performs better than OHEM(used in original SSD). Thank you!

gakkiri commented 5 years ago

For me, focal loss didn't work well in SSD.

zhaohao0404 commented 4 years ago

I have a question : why exclude the number of negative sample in multi_box_loss? (Line 68 69)