search

chuanqi305 / MobileNet-SSD

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
MIT License
2.04k stars 1.18k forks source link

focal-loss #15

Open kaishijeng opened 7 years ago

kaishijeng commented 7 years ago

Do you plan to add focal loss to boost SSD performance? https://arxiv.org/abs/1708.02002

Thanks,

chuanqi305 commented 7 years ago

Thank you for your suggestion, now I'm doing this job.

kaishijeng commented 7 years ago

Liu Y. has released focal-loss layer in this repository below and it may save your time to reinvent the same thing: https://github.com/sciencefans/Focal-Loss

kaishijeng commented 7 years ago

Any luck with focal-loss training?

Thanks,

chuanqi305 commented 7 years ago

@kaishijeng My version of focal loss for SSD: https://github.com/chuanqi305/FocalLoss Unfortunately, the performance has not improved.

kaishijeng commented 7 years ago

I will take a look at your focal loss implementation. Any idea why the performance is not improved. According to the paper, it helps a lot of hard class.

Thanks