decouples / SSD-TF

SSD: Single Shot MultiBox Detector in TF

SSD-TF

SSD: Single Shot MultiBox Detector in TF

Usage

• Download Pascal VOC Dataset and reorganize the directory as follows:

//or get from here
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar  
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar  
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar  

VOCROOT/
	   |->VOC2007/
	   |    |->Annotations/
	   |    |->ImageSets/
	   |    |->...
	   |->VOC2012/
	   |    |->Annotations/
	   |    |->ImageSets/
	   |    |->...
	   |->VOC2007TEST/
	   |    |->Annotations/
	   |    |->...

VOCROOT is your path of the Pascal VOC Dataset.

• Run the following script to generate TFRecords.

   python dataset/convert_tfrecords.py --dataset_directory=VOCROOT --output_directory=./dataset/tfrecords

• Download the pre-trained VGG-16 model (reduced-fc) from here and put them into one sub-directory named 'model' (we support SaverDef.V2 by default, the V1 version is also available for sake of compatibility).

• Run the following script to start training:

   python train_ssd.py

• Run the following script for evaluation and get mAP:

   python eval_ssd.py
   python voc_eval.py

  Note: you need first modify some directory in voc_eval.py.

• Run the following script for visualization:

   python simple_ssd_demo.py