This article aims to help out beginners in machine learning on creating your own custom object detector. I have been trying to create a simple object detector and had to go through many articles spread across the internet to find all the required information. So I figured I’ll gather all the information I found in one place to make things easier for the next me.I’ll keep this as easy and informative as possible.
This article describes everything required to create a working object detector from gathering data to exporting the model for our use.
- Prerequisites
- Setting up the work environment
- Making the dataset
- Downloading and configuring the pre-trained model
- Training and evaluation
- Exporting the model
Prerequisites
This tutorial does not assume any previous knowledge of TensorFlow. I have tried to keep it as simple as I can so that anyone could get a working model at the end. For beginners, I definitely suggest this exercise as I found it to be an excellent first step into the world of transfer learning.
This tutorial utilizes Python. Several python packages are required to get this going. I’m just going to list them below. If you are reading this, you’ll probably have them already installed. If not, all these packages are very popular and there are lots of tutorials on the internet on how to install them.
- Pandas
- Tensorflow
- Tensorboard (Optional)
- Openimages
#machine-learning #data-science #transfer-learning #artificial-intelligence