Skip to content

Running object detection on a webcam feed using TensorRT on NVIDIA GPUs in Python.

Notifications You must be signed in to change notification settings

NVIDIA/object-detection-tensorrt-example

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Object Detection TensorRT Example:

This python application takes frames from a live video stream and perform object detection on GPUs. We use a pre-trained Single Shot Detection (SSD) model with Inception V2, apply TensorRT’s optimizations, generate a runtime for our GPU, and then perform inference on the video feed to get labels and bounding boxes. The application then annotates the original frames with these bounding boxes and class labels. The resulting video feed has bounding box predictions from our object detection network overlaid on it. The same approach can be extended to other tasks such as classification and segmentation.

A detailed explanation of this code sample can be found in the How to Apply Deep Learning for Common Applications webinar and also as a blog on the NVIDIA Medium Page.

1. Setup the environment

./setup_environment.sh

2. Run inference on the webcam

python SSD_Model/detect_objects_webcam.py 

Note: By changing the argument to the "p" flag, you can change which precision the model will run in. The options are FP32 (-p 32), FP16 (-p 16), and INT8 (-p 8)

Note: When running in INT8 precision, an extra step will be performed for calibration. But just like building an engine, it will only be performed the first time you run the model.

About

Running object detection on a webcam feed using TensorRT on NVIDIA GPUs in Python.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published