目录
前言:之前用YOLOv5训练了一个口罩检测模型,但是想让模型在k210上运行,涉及到模型的转换,下面我就我个人转化历程来简单记录一下,我的艰辛转换历程(如有错误的话,就评论区指出来,防止我误人子弟)
第一种:用yolov5训练模型之后在进行模型转换 ,我的转换思路是pt–>onnx–>tflite–>kmodel,转化过程异常艰辛,中间有各种各样的坑,我不出意外果然出了意外,在最后一步时失败了,小白不建议走这条路,如果你硬要逝世的话,下面给你一些参考博客链接
1.模型转换:pytorch模型转onnx, onnx转tensorflow, tensorflow转tflite_城俊BLOG的博客-CSDN博客_onnx转tensorflow
最后一步tflite转kmodel用NNcase这个软件就行.
第二种: 用maixpy官方线上训练网站—maixhub,进行线上训练,只需要你把数据集按一定格式放上去训练,训练结果直接是kmodel(可以在k210上跑),至于教程嘛maixhub上有视频教程,附:网站链接:MaixHub
第三种:用Mx-yolov3来进行训练,嘿嘿,用过1.2版本的如果环境能配置好的话,还行挺好用(你只要把你数据集放上去,训练出来就是tflite,而且安装mx-yolov3会把nncase直接给你装上去,所以基本是一套龙服务了,巴适的很!),如果按教程配不好的话,基本G了,这里推荐一篇博客:MX-Yolov3本地训练模型的安装和使用_BYEBlackBird的博客-CSDN博客_yolov3安装
第四种: 还是用mx-yolov3,不过是3.0版本,相比于1.2版本3.0版本不需要配置环境!不需要配置环境!不需要配置环境!非常好用,你只需要把数据集正好就行,但是也有一些坑,下面是我遇到的坑,1.照片或者是xml名字不能有空格(可能下划线也会被视为空格,反正尽量数字或者字母)2.照片分辨率有要求,我一直用的是224*224,我爬的照片没有统一分辨率,就一直报错,具体要求是什么我不太清楚,是必须正方形?还是统一就行?还是是224倍数就行?我不太了解懒得一个个试,mx-yolov3里有一个叫imagetool的程序可以把照片统一成224*224的规格,比较方便,使用方法和上面1.2版本教程一样,跳过配置环境步骤即可!下面是3.0版本的百度网盘链接:https://pan.baidu.com/s/1Gl3Qfw5s8LZuu2wc1GTITg 提取码:dvsf
结语:希望可以帮助到大家,如果有什么错误之处,望各位指出,大家一起进步!同时也感谢上面被我引用博客的博主!
文章出处登录后可见!
已经登录?立即刷新