TensorFlow Object Detection API tutorial

Important

This tutorial is intended for TensorFlow 1.14, which (at the time of writing this tutorial) is the latest stable version before TensorFlow 2.x.

Tensorflow 1.15 has also been released, but seems to be exhibiting instability issues.

A version for TensorFlow 1.9 can be found here.

At the time of righting this tutorial, Object Detection model training and evaluation was not migrated to TensorFlow 2.x (see here). From personal tests, it seems that detection using pre-trained models works, however it is not yet possible to train and evaluate models. Once the migration has been completed, a version for TensorFlow 2.x will be produced.

This is a step-by-step tutorial/guide to setting up and using TensorFlow’s Object Detection API to perform, namely, object detection in images/video.

The software tools which we shall use throughout this tutorial are listed in the table below:

Target Software versions
OS Windows, Linux
Python 3.7
TensorFlow 1.14
CUDA Toolkit 10.0
CuDNN 7.6.5
Anaconda Python 3.7 (Optional)

Indices and tables