K210板教程6——模型训练

2020-12-241.8万
AI 快速预览详细 收起
本教程介绍了如何使用袁老师的软件训练YOLO模型。首先,通过爬虫工具获取并处理图片,然后对图片进行标注。接着,使用袁老师的软件进行模型训练,最后将训练好的模型转换为kmodel格式。整个过程简单易懂,无需复杂的编程知识,小学生也能独立操作。


通过前面的学习,我们会发现只要有训练好的模型,博派可以识别任何物体。
那这些模型从何而来呢?
打开https://www.maixhub.com/
这是Maixduino官方的一个模型网站,里面有很多现成的模型可以下载到博派的板子里直接用。


K210板教程6——模型训练图1



该网站暂时可以下载的模型不多,但是该网站支持个人训练的模型上传
随着使用的人越来越多,相信未来会有很多模型可以在该网站上下载。
该网站还支持在线训练模型,只要你准备好自己的图片集和标注集,上传之后它就会帮你训练好,然后你下载好训练的模型就可以直接用了。


K210板教程6——模型训练图2



虽然可以找到别人训练的模型,但是有时候无法满足我们的个性需求,这时候就要求我们自己训练模型。
但是模型该如何训练呢?
本节就来讲如何训练模型,这里要请出一个超级大神-广东省中山市三乡镇新圩小学袁运强老师。


K210板教程6——模型训练图3



就是上图中最右边的很帅的老师。
他做的事情也非常帅。
原本YOLO训练模型是一件很难的事情,要装很多的Python库,要用代码进行,这对于很多人来说难度太大了,而袁老师将这一切工作做成了一个软件,只需要点点鼠标就可以训练YOLO模型,小学生都可以独立操作,这实在是太酷了。
袁老师在论坛有发帖,介绍他的软件如何训练模型,大家可以移步去看看。
https://mc.dfrobot.com.cn/thread-307554-1-1.html
软件的获取:
大家关注袁老师微信公众号“Import 创客”,回复:“Mx3“即可获得下载链接。
下面我将详细的为大家介绍如何利用袁老师的软件训练模型。
步骤1获取图片



我们要训练一个模型,必须给软件足够多的图片让其学习。
以口罩识别为例,我们必须准备最少60张以上戴口罩(最好各种颜色各种款式的口罩都有)的图片和至少60张没戴口罩的图片。
然后对这些图片进行标注,最后让机器去学习,图片越多识别率越高,当然工作量也越大。
获取图片的方式有两种:
第一种:自己用手机或相机拍摄。
第二种:上网下载。
第一种我们这里不讲,重点讲讲第二种。
袁老师的软件里有一个专门上网爬图片的软件,可以直接拿来用。


K210板教程6——模型训练图4



K210板教程6——模型训练图5



设置好关键词,保存位置,数量点击下载即可。
另外我自己也做了一个小工具,既然袁老师有了为什么我还要做呢?
因为图片下载后还要进行处理,我的小软件就是处理这个的。
我的小工具比较简陋,一切以实用为主。文章最后有下载
打开如图所示


K210板教程6——模型训练图6



点击爬虫,打开软件如下,用法和袁老师的软件一样,只是多了一个从哪个网站爬取图片,默认是百度。


K210板教程6——模型训练图7



要注意爬取图片的时候关键词设置很重要,以苹果为例,它会将苹果手机也爬下来,因此我们要设置多个关键词。比如水果 苹果


K210板教程6——模型训练图8



设置好关键词之后,再看,这才是我们需要的


K210板教程6——模型训练图9

步骤2处理图片

图片准备好了,还不能就这样直接给软件去学习,需要对图片进行处理。
第一:处理好文件名。
第二:处理好大小。
我们要将所有的图片都放到同一个文件夹下,因此我们需要对图片进行改名。


K210板教程6——模型训练图10



新建一个文件夹命名为images,将香蕉和苹果的图片重命名之后,一起复制到images文件夹下。


K210板教程6——模型训练图11



接着要将这些图片设置好大小,YOLO训练的话最好是正方形,建议设置为224*224
太大了训练起来非常慢。


步骤3标注图片



图片处理好之后,接下来就要标注图片了。
新建一个文件夹fruit,然后将images文件夹复制进去。在fruit下新建一个文件夹xml
打开袁老师的软件,找到lableimg


K210板教程6——模型训练图12

K210板教程6——模型训练图13



第一步:更改图片所在位置。
第二部:更改标签所在位置。


K210板教程6——模型训练图14



第三步:标注图片
第四步:保存标注
第五步:下一张


K210板教程6——模型训练图15



步骤4开始训练



所有图片都标注好了,就可以开始训练了。
打开袁老师的软件,先择物体识别,具体操作步骤如下:


K210板教程6——模型训练图16



训练的时间几个小时到几十个小时不等,根据电脑性能而定,建议用GPU进行训练。
训练好之后的模型文件是这样的


K210板教程6——模型训练图17



我们还需要转换一下,在刚才训练模型的界面,有一个模型转换


K210板教程6——模型训练图18



设置好模型的路径和保存的路径,量化图片路径选择images文件夹即可,点击转换
几分钟之后就生成了一个kmodel模型,这就是我们需要的。
至此模型训练完成。
最后再次感谢袁老师,为人工智能教育普及所做的贡献。

如果你的机器比较慢,也可以将我们整理好的图片集和标注集发送到https://www.maixhub.com/mtrain.html,借助其强大的机器算力帮我们训练。
附件

创作许可协议

本项目采用 None(不开放任何权利,保留所有权利) 进行许可。

评论(12)
花生编程
花生编程
不错不错
花生编程
花生编程
厉害厉害
三春牛-创客
三春牛-创客
赞赞赞赞赞
三春牛-创客
三春牛-创客
厉害厉害
peterlhj
peterlhj
请问大佬下面图片的报错是什么原因?
黑猫警长
黑猫警长
请教一下两个k.model模型可以合并成一个嘛?maixhub一次只能上传20M训练,想做一个大概一百种物体检测,需要分几次训练
温暖的蓝色
温暖的蓝色回复黑猫警长
选择本地训练吧,网站是有数量限制的
郝仁
郝仁
你好大佬,我刚好看到你的这一篇博客,对那个批量转换图片的工具很感兴趣,但它不能下载了,不知能不能分享一下其他可以下载的连接
肥罗-阿勇
肥罗-阿勇
作者
回复郝仁
文章最好那个附件点击就可以下载了,我刚试了都可以的
郝仁
郝仁
想问下,这个软件打开之后的界面怎么缩小,信息和终端那部分都看不到
王尧坤
王尧坤
还没尝试,有机会尝试下,看看能不能成功
dadabo2000
dadabo2000
您好,请问我按所有步骤完成数据集、标注、参数设置,在开始训练后,总是出现系统没有找到文件?
人生莫回忆
人生莫回忆
想问一下您有没有遇到过模型转换失败的情况
DFByc2ee9Hd
DFByc2ee9Hd回复肥罗-阿勇
您好,我也出现这个问题,请问怎么发给你模型,我的q是504577506.
DFByc2ee9Hd
DFByc2ee9Hd回复肥罗-阿勇
你好,我也出现了转换不了的情况,请问怎么联系你。
共3条回复,已加载2条,点击查看更多共3条回复,已加载全部
王尧坤
王尧坤
膜拜,太厉害了,学习了很多新知识
- 没有更多了 -