README.md

Nanodet PyTorch=>ONNX=>TensorRT

1.Reference

• nanodet: https://github.com/RangiLyu/nanodet
• get nanodet pretrained weights from here: COCO pretrain weight for torch>=1.6(Google Drive) | COCO pretrain weight for torch<=1.5(Google Drive)

2.Export ONNX Model

git clone https://github.com/linghu8812/nanodet.git

copy export_onnx.py into nanodet/tools and run export_onnx.py to generate nanodet-m.onnx.

python tools/export_onnx.py

3.Build nanodet_trt Project

mkdir build && cd build
cmake ..
make -j

4.run nanodet_trt

./nanodet_trt ../config.yaml ../samples

5.Results: