1. 从百度下载TensorRT-5.0包:
链接:https://pan.baidu.com/s/1buOjyLE5DPh6yLKK8e9Mrg
提取码:834t
2.解压包,然后cd至/TensorRT-5.0.2.6/targets/x86_64-linux-gnu/samples/python/yolov3_onnx。
3.配置安装环境:请参考tensorrt环境配置
4.yolov3.weight转yolov3.onnx模型:
a)将with open(cfg_file_path, 'rb')的b干掉。如下图所示:
b)找到yolov3_to_onnx.py,然后干掉下图的代码:
c)结合实际情况来修改yolov3.weigth、yolov3.cfg、yolov3.onnx的路径。
d)如果输入图片大小是608,则下图所示:
如果输入图片大小是416,则如下图所示:
注意:相应的yolov3.cfg也需要修改相关参数。
e)直接python yolov3_to_onnx.py
5.生成yolov3.onnx模型后,继续找到onnx_to_tensorrt.py。
a)如果输入图片大小是416,就如图所示进行修改
b)python onnx_to_tensorrt.py