Name		Name	Last commit message	Last commit date
parent directory ..
CMakeLists.txt		CMakeLists.txt
README.md		README.md
export_onnx.py		export_onnx.py
inference.py		inference.py
lenet.py		lenet.py
requirements.txt		requirements.txt
train.py		train.py

README.md

LeNet Network PyTorch=>ONNX=>TensorRT

1.Reference

LeNet: Gradient-Based Learning Applied to Document Recognition

2.Train PyTorch Network

python3 train.py

3.Export ONNX Model

python3 export_onnx.py

4.Build lenet_trt Project

cd ../  # in project directory
mkdir build && cd build
cmake ..
make -j

5.Run lenet_trt

cd ../../bin/
./tensorrt_inference lenet ../configs/lenet/config.yaml ../samples/mnist

6.Benchmark(2080Ti)

model	PyTorch	TensorRT
inference time	0.4ms	0.046ms