nanodet 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 &
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/119743422