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

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
female		female
male		male
model		model
util		util
.demo.png		.demo.png
.gitignore		.gitignore
CNNClassificationTensorRT.py		CNNClassificationTensorRT.py
LICENSE		LICENSE
README.md		README.md
demo.py		demo.py