Training a Machine Learning model for custom object detection or custom object image classification
About the training process
IMAGE CLASSIFICATION vs OBJECT DETECTION
In Image Classification, we have one label per image while for object detection, we can have single or multiple labels per image as demonstrated below.
TRAINING AN ML MODEL FOR OBJECT DETECTION
We can use either of the TF1 or TF2 model zoo’s to select our model for training.
Here are 2 tutorials on how to train a machine learning model for custom object detection for both the TensorFlow versions (TF 1.x & TF 2.x) using Google colab.
TRAIN AN ML MODEL FOR IMAGE CLASSIFICATION
Here are 2 tutorials on how to train a model for custom object image classification using Google colab and Teachable Machine.