Wameh / object_detection_lab

Leverage existing tensor flow models to detect objects in images and videos

Object Detection

Leverage existing tensor flow pre-trained models to detect objects in images and videos

Object detection under various weather conditions during driving

Another example to detect objects in an indoor office space

Instruction

• Download the pretrained models from the model zoo.
• Make sure the update the model file path in object_detection.py

Usage

Run python object_detection.py --help for available options

example:

• python object_detection.py --type image --image_file examples/dog.jpg
• python object_detection.py --type video --video_file examples/driving.mp4

Examples of results

SSD (Single Shot MultiBox Detection) based on MobileNet

Original Image	Result

R-FCN based on ResNet-101

Original Image	Result

Faster R-CNN based on Inception-ResNet-v2

Original Image	Result