文章作者：激萌の小宅

促销：￥0

价格：￥0

配送方式：购买后立即生效（如购买异常，请联系站长）

付款之后一定要等待自动跳转结束，否则购买可能会失败

0 天

有效期
0

总销量
0

累计评价

详细说明

<h1 style="font-size: 32px; font-weight: bold; border-bottom: 2px solid rgb(204, 204, 204); padding: 0px 4px 0px 0px; text-align: left; margin: 0px 0px 10px;">RDK X5 yolov5目标检测之pt转onnx</h1>接上一篇博客 参考博客：<a href="https://developer.d-robotics.cc/forumDetail/163807123501918330" target="_blank">https://developer.d-robotics.cc/forumDetail/163807123501918330</a>打开最外层目录的export.py来到主函数下，修改刚刚训练出的模型的路经，特别要注意的是这边的参数--img-size最好是32的倍数，且一定要记得你模型尺寸（此参数） <img class="pimg" src="http://www.bilibili996.com/upload/image/638d22fd95584e01ba80b412dd70b2f3.jpg?638647718024996653" title="QQ20241017-142314.jpg" alt="QQ20241017-142314.jpg"/> 继续看“# ONNX export”开始的部分，opset_version就是转化后的ONNX版本，默认12,务必要改成11！！！否则模型转bin文件会报错；此外，这边的input_name也是你需要记得的【备注：yolov5_v2.0_leaf工程已经将这部分帮我们修改好了，如果你采用的是原始工程，请对照着修改】<img class="pimg" src="http://www.bilibili996.com/upload/image/346c595fe8aa4147af2e854ffc494fe8.jpg?638647719948096378" title="QQ20241017-142626.jpg" alt="QQ20241017-142626.jpg"/> <span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">来到<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">models/yolo.py</code><span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">第22行的<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">forward</code><span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">方法中，把29行的<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">x[i] = x[i].view...(省略)...contiguous()</code><span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">注释掉，替换成<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">x[i] = x[i].permute(0, 2, 3, 1).contiguous()</code>【备注：yolov5_v2.0_leaf工程已经将这部分帮我们修改好了，如果你采用的是原始工程，请对照着修改】<img class="pimg" src="http://www.bilibili996.com/upload/image/1cc8a269ae9549eb8a21f6cb2060d1d4.jpg?638647723858573819" title="QQ20241017-143255.jpg" alt="QQ20241017-143255.jpg"/> <span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">保存修改后直接运行<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">export.py</code><span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">，就可以在<code style="margin: 0px; padding: 0.065em 0.4em; box-sizing: border-box; font-size: 0.87em; font-family: Menlo, Monaco, Consolas, "Courier New", monospace; background-color: rgb(255, 245, 245); border-radius: 2px; color: rgb(255, 80, 44); overflow-x: auto; word-break: break-word;">runs/exp0/weights/</code><span style="color: rgb(37, 41, 51); font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "Helvetica Neue", Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol"; font-size: medium; background-color: rgb(255, 255, 255);">中看到onnx文件啦<img class="pimg" src="http://www.bilibili996.com/upload/image/dec5b1dc0b1c42b4b2a5fe8e107adfd6.jpg?638647728764601145" title="QQ20241017-144110.jpg" alt="QQ20241017-144110.jpg"/>

RDK X5 yolov5目标检测之pt转onnx

接上一篇博客

参考博客：https://developer.d-robotics.cc/forumDetail/163807123501918330

打开最外层目录的export.py来到主函数下，修改刚刚训练出的模型的路经，特别要注意的是这边的参数--img-size最好是32的倍数，且一定要记得你模型尺寸（此参数）

继续看“# ONNX export”开始的部分，opset_version就是转化后的ONNX版本，默认12,务必要改成11！！！否则模型转bin文件会报错；此外，这边的input_name也是你需要记得的

【备注：yolov5_v2.0_leaf工程已经将这部分帮我们修改好了，如果你采用的是原始工程，请对照着修改】

来到models/yolo.py第22行的forward方法中，把29行的x[i] = x[i].view...(省略)...contiguous()注释掉，替换成x[i] = x[i].permute(0, 2, 3, 1).contiguous()

【备注：yolov5_v2.0_leaf工程已经将这部分帮我们修改好了，如果你采用的是原始工程，请对照着修改】

保存修改后直接运行export.py，就可以在runs/exp0/weights/中看到onnx文件啦