How To Train Your Own SSD Object Detection Model(Based on Tensorflow)?

Source: Deep Learning on Medium

How To Train Your Own SSD Object Detection Model(Based on Tensorflow)?

Anyone who’s interested in training SSD model can refer to the following link:

The steps:

  1. Install LabelImg
  2. Using LabelImg to annotate photos
  3. Config SSD
  4. Download pre-trained model
  5. Train model
  6. Use the model to predict

After reading the article, I found that the training process of SSD is similar to YOLOv3. They both use LabelImg to annotate photos but with different file type.

For SSD is VOC (*.xml);for YOLOv3 is YOLO(*.txt)