AX620A转换yolo系列模型踩坑

文章介绍了在AX620A平台上转换YOLOV5和YOLOV8模型时遇到的问题及解决方案。YOLOV5转换需要注意pytorch版本不能高于1.13.0,并在使用onnx-modifier时勾选shapeinference和cleanup选项。而YOLOV8转换的关键是使用python脚本直接导出模型,因原有导出命令行工具可能未更新。转换过程在12代i5CPU上耗时较长,对多线程支持不佳。

AX620A转换yolo系列模型踩坑


前言

去年记录了如何在ax620a下使用工具链转换自训练yolov5模型:AX620A运行yolov5s自训练模型全过程记录
但今天由于各种环境发生变化了,再去这样转换的话,就会报错失败。
在这里插入图片描述
经过一番查找,发现是pytorch版本更新导致的。其实社区有文章已经更新了yolov5和yolov8的模型转换方法,但是仍然有坑点,如果不注意的话还是会失败。


一、YOLOV5转换

社区文章看这里:[m3axpi] YOLOv5训练到部署全流程
其实主要问题就是pytorch版本不能大于1.13.0。如果是使用1.13.0之后版本的话,按照文章第二种方法,使用onnx-modifier图形化工具删去后处理算子。但在编辑完成下载新模型时,一定要勾选shape infrernceclean up两个选项,再download导出,否则转换时还是会报错失败。
在这里插入图片描述

二、YOLOV8转换

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值