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. Using LabelImg to annotate photos
  2. Config SSD
  3. Download pre-trained model
  4. Train model
  5. 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)

Written by

Machine Learning / Deep Learning / Python / Flutter

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store