Detect different objects in an image with TensorFlow
Inference Graph is in zip file frozen_inference_graph.zip
Click on this image to se how object detection works:
- Put an image in test_images folder
- Then run the object_detection.ipynb file to get results
You can use different pre-trained model like for SSD,R-CNN, Faster-RCNN
But here i have used ssd
Read more about SSD here
- Get Ground-Truth with labelImg which helps you to draw bounding box and save them as json files
- Then you run result.py to know how your model has performed
Created by:
Anubhav Shukla
This application of object detection is also given in the official github page of tensorflow for object detection.
