yolov6模型转换方法

人工智能领域优质创作者

博客专家认证

2023-01-13 03:14:57

课时名称	课时知识点
yolov6模型转换方法	使用tensorrt工具，将pytorch版本的模型转换为tensorrt格式的模型

...全文

142 回复打赏收藏转发到动态举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

安装onnx，转的时候提示出错ONNX: export failure 0.1s: Unsupported ONNX opset version: 17。yolov5本身release目录有提供了onnx转换好的模型，想着也自己操作一遍，可是实际操作却遇到了问题，这里做下记录方便后续可能用到。修改export.py里的默认设定如下，原先默认是17，改为12后可以正常转，会在目录生成转换的onnx模型。验证转换后的onnx模型时会提示出错，按提示安装pip install onnxruntime。

YOLOv5模型转换ONNX格式工具：加速模型部署的利器【下载地址】yolov5转换ONNX格式工具本仓库提供了`yolov5_convert_onnx.zip`资源文件，旨在帮助用户方便地将YOLOv5模型从PyTorch(.pt)格式转换为ONNX格式。ONNX（Open Neural Network Excha...

本文介绍了如何将YOLOv8模型从PyTorch格式转换为ONNX格式，并进行优化和验证。

用下面板载的程序，这个程序修改于rknn-toolkit2-1.4.0中的examples/onnx/yolov5/text.，测试一下,使用刚转换的best.rknn模型，推理名为images4.jpg的照片。修改rknn-toolkit2-1.4.0/examples/onnx/yolov5/text.py 文件，ONNX_MODEL为刚生成的.onnx文件，RKNN_MODEL为将要转换成.rknn的文件。训练好后导出.rknn模型，best.pt 文件在./run/中。

linux下pythorch训练yoloV5模型转换为tensorRT模型

宗孝鹏的课程社区_NO_1

4

社区成员

11

社区内容

发帖

与我相关

我的任务

社区管理员

加入社区

近7日
近30日
至今

加载中

查看更多榜单

社区公告

暂无公告

试试用AI创作助手写篇文章吧

+ 用AI写文章