/TensorFlow2.0_SSD

A tensorflow_2.0 implementation of SSD (Single Shot MultiBox Detector) .

Primary LanguagePythonMIT LicenseMIT

TensorFlow2.0_SSD

A tensorflow_2.0 implementation of SSD (Single Shot MultiBox Detector) .

Requirements:

  • Python >= 3.6
  • TensorFlow == 2.1.0
  • numpy == 1.17.0
  • opencv-python == 4.1.0.25

Usage

Train on PASCAL VOC 2012

  1. Download the PASCAL VOC 2012 dataset.
  2. Unzip the file and place it in the 'dataset' folder, make sure the directory is like this :
|——dataset
    |——VOCdevkit
        |——VOC2012
            |——Annotations
            |——ImageSets
            |——JPEGImages
            |——SegmentationClass
            |——SegmentationObject
  1. Run write_voc_to_txt.py to generate voc.txt.
  2. Run train.py to start training, before that, you can change the value of the parameters in configuration.py.

Test on single picture

  1. Change the test_picture_dir in configuration.py.
  2. Run test.py to test on single picture.

References