Focal_loss_for_object_detection
Focal_loss_for_object_detection copied to clipboard
Retinanet impelemented by Pytorch : 69.0 mAP on VOC2007 test.
Train RetinaNet with Focal Loss in PyTorch.
Reference:
[1] Focal Loss for Dense Object Detection
Imporvements:
Add the eval.py to calculate mAP on VOC.
Support batch_size>1 during test.
Add the produce_txt.py to produce train.txt and val.txt for voc data.
Results: mAP: 69.0 points on VOC2007_trainval datasets.