search

yhenon / pytorch-retinanet

Pytorch implementation of RetinaNet object detection.
Apache License 2.0
2.14k stars 665 forks source link

mAP stuck while training on VisDrone #165

Open mahmedraza opened 4 years ago

mahmedraza commented 4 years ago

VisDrone is apparently a small dataset. As, I could perceive from these issues already opened is that retinanet does not perform well in the case of small datasets (Is there any workaround for it?). However, [https://arxiv.org/abs/1906.00786](this paper) shows retinanet achieves almost 29% mAP for VisDrone. Whereas, my mAP (using this repo) gets stuck on 10.8% even after 75 epochs. Is there any bug in this repository? Or if anyone else has got different results for VisDrone dataset with this repo? Any help would be appreciated.

efedoganay commented 4 years ago

I was able to get somewhere around 15%. Still not good though. I recommend you to play with anchor settings.