README.md

CNNClassificationTensorRT

Python implementation of using CNN model with TensorRT

Tested using:

Ubuntu 18.04
Python 3.6.8
CUDA 10.1
CuDNN 7.6
TensorRT 5.1.x
darkflow
tensorflow-gpu 1.14.z
OpenCV 4.0.1

Ryzen 5 1600
Nvidia GTX 1070

Demo

python3 demo.py

100 images in 67ms