tensorrt模型转换及优化

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