Open In Github
TF2.0 Object Detection Inference
Python program to perform object detection inference with Tensorflow 2.0.
Installation
- Follow the steps as in: https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html
Usage
Modify the model name (model_name) with the desired model from the Object Detection Model Zoo (TF2). Note: The name should be the one in the link, for example, in the case of the SSD MobileNet v2 320x320 model, model_name should be ‘ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8’.
Modify the relative path to the labels folder (PATH_TO_LABELS). The current program works only if the Python program is in the same directory where the models directory is located.