TensorFlow 2 Object Detection API tutorial

Important

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

A version for TensorFlow 2.2 can be found here.

A version for TensorFlow 1.14 can be found here.

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.9 1

TensorFlow

2.5.0

CUDA Toolkit

11.2

CuDNN

8.1.0

Anaconda

Python 3.8 (Optional)

1

Python 3.9 is not a strict requirement. Any Python 3.x version should work, although this has not been tested.