Name		Name	Last commit message	Last commit date
parent directory ..
samples		samples
CMakeLists.txt		CMakeLists.txt
README.md		README.md
coco.names		coco.names
config.yaml		config.yaml
export_onnx.py		export_onnx.py
main.cpp		main.cpp
nanodet.cpp		nanodet.cpp
nanodet.h		nanodet.h
prediction.jpg		prediction.jpg

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: