Skip to content

Object Detection Model on TF2

Latest
Compare
Choose a tag to compare

Object Detection with TF2

This release contains the necessary files to run the Object Detection Model with Tensorflow 2.

Usage

To use this model, download .zip file and add it under /object_detection/models.

import os
MODEL_NAME = 'saved_model'
CWD_PATH = './models/model_tf2'
PATH_TO_SAVED_MODEL = os.path.join(CWD_PATH, MODEL_NAME)
PATH_TO_LABELS = os.path.join(CWD_PATH,'label_map.pbtxt')

Example from get_objects_and_coordinates_in_terminal_tf2.py
Path (currently in branch vision/_object_detection_tf2): Robocup-Home/object_detection/scripts/