Skip to content

Latest commit

 

History

History
35 lines (22 loc) · 994 Bytes

README.md

File metadata and controls

35 lines (22 loc) · 994 Bytes

Single-Shot Refinement Neural Network for Object Detection

Abstract

This is a MxNet_gluon re-implementation of Single-Shot Refinement Neural Network for Object Detection.

This project is based on MxNet_GluonCV, and completed by YangJirui.

Train on VOC 2007+2012 trainval and test on VOC 2007 test.

Comparison

use voc2007_metric

SSD 300(vgg-16): 77.2 mAP
Paper's refineDet 320(vgg-16): 80.0 mAP
Our refineDet 320(vgg-16) : 78.9 mAP

Our re-implementation of RefineDet performs about 1mAP lower than Paper's.
We will try our best to fix it in the future.

My Development Environment

1、python2.7 (anaconda recommend)
2、cuda9.0 (cuda 8.0 may cause Nan when training)
3、opencv(cv2)

Train

1、train

cd $PATH_ROOT/scripts/detection/refineDet
./train_scipt.sh