ycdhqzhiai / nanodet_tensorrt

nanodet-plus-m 416 tensorRT实现

DanoDet-Plus TensorRT

1.生成wts文件

• 将gen_wts.py文件copy到nandeodet原始仓库
• 选取合适参数(config文件和pth模型),运行脚本得到wts文件

2.生成engine文件运行

mkdir build && cd build
make -j4
./nanodet -s
./nanodet -d

3.Reference resources

• 1.https://github.com/wang-xinyu/tensorrtx
  
• 2.https://github.com/RangiLyu/nanodet
  
• 3.https://docs.nvidia.com/deeplearning/tensorrt/api/c_api