nanodet

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

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

4.run nanodet_trt

cd ../../bin/
./tensorrt_inference nanodet ../configs/nanodet/config.yaml ../samples/detection_segmentation

5.Results: