The object detection model is based on SSD-InceptionV2 model, pretrained on the COCO Detection task.
The model definition is from - Tensorflor Object Detection API
This repository contains the:
- Model definition config
- Class label proto
- Object Prediction code
sys.path.append("path-to-tf-models-object-detection-folder") Update TEST_IMAGE_PATHS Update output_path
Run using :
python object_prediction.py
The file output will be stored in
detections_50.json