Tensorrt-Python-Inference-for-Caffe-SSD

This code provides a python wrapper for inference of SSD caffe models. Change the prototxt file by following these instructions and run caffe_ssd.py. In the code change all_classes variable to your required classes.

https://docs.nvidia.com/deeplearning/tensorrt/release-notes/tensorrt-5.html#rel_5-0-2

Tensorrt version: 7.1

Ubuntu 18.04

Cuda 10.2

Python 3.6

OpenCV: 4.1