ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 167 forks source link

COCO2017-yolov3-map error #78

Closed QZ-cmd closed 3 years ago

QZ-cmd commented 4 years ago

I use yolov3-MobilenetV1 Pre training weight to train 100 epoch,but map only 46.4%(iou=0.5),I want to know what's wrong. thanks

ww-zwj commented 4 years ago

I think the author has done pre-training on coco and then directly training on VOC, so the total training epochs are only 100. If you train directly on VOC, you need to increase the total epoch and Warm UP Epochs, and increase the initial learning rate.

QZ-cmd commented 4 years ago

I think the author has done pre-training on coco and then directly training on VOC, so the total training epochs are only 100. If you train directly on VOC, you need to increase the total epoch and Warm UP Epochs, and increase the initial learning rate.

Thanks for your reply