In the following codes, we will load a pre-trained model and perform an object detection task on pre-defined 90 categories. Before we begin, check that there is a working TensorFlow (Version = 1.4.0) Python environment.
In this section, our input is as shown in the image with people:
We will follow similar instructions as that of two-stage detectors and begin by cloning TensorFlow/models repo:
git clone https://github.com/tensorflow/models.gitcd models/research
Let's download a pre-trained model from TensorFlow model-zoo. These are for one-stage detectors:
- For macOS X:
curl -O http://download.tensorflow.org/models/object_detection/ssd_inception_v2_coco_2017_11_17.tar.gz ...