Home tf2 object detection inference
Post
Cancel

Open In Github

TF2 Object Detection Inference

TF2.0 Object Detection Inference

Python program to perform object detection inference with Tensorflow 2.0.

Installation

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.

This post is licensed under CC BY 4.0 by the author.