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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容