TensorFlow 2 Object Detection API tutorial

Important

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

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.8

TensorFlow

2.2.0

CUDA Toolkit

10.1

CuDNN

7.6.5

Anaconda

Python 3.7 (Optional)

Indices and tables