pursu / Stronger-One-stage-detector-with-much-Tricks

Stronger SSD & YOLO v3

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Stronger One-stage detector with much Tricks

This repo was inspired by the paper Bag of Freebies for Training Object Detection Neural Networks.

I would test popular training tricks as many as I can for improving one-stage detector accuarcy, feel free to leave a comment or email me about the tricks you want me to test (yhao.chen0617@gmail.com).

Traing Data : VOC0712 trainval

Test data : VOC07 test

GPU : TITAN X (pascal)

Framework : Pytorch 0.4

Network mAP FPS Parameter
SSD 300 79.62 ~100 -
YOLOV3 544 - - -

Note:

  • Stronger YOLOv3 with much tricks will be released soon.
  • This repo does not use multi-scale train because of the limitation of GPU memory (I only have one card), which is extremely beneficial to detector.

About

Stronger SSD & YOLO v3


Languages

Language:Python 91.3%Language:C 5.2%Language:Cuda 2.5%Language:Shell 0.9%Language:C++ 0.1%