我的准备
- Maix duino开发板一块(含摄像头配件)
- Type-c数据集一根
垃圾的图片数据集
分四类垃圾:厨余垃圾、有害垃圾、可回收垃圾、其它垃圾。每类垃圾分一个文件夹,文件夹中就是该类垃圾的图片。
MaixPy IDE(软件)
编辑用来调用模型的代码
kflash_gui(软件)
将模型文件烧录(下载)到开发板上
数据集:有许多地方可以下载,例如:
中文生活垃圾分类数据集-modelscope
百度飞桨-AIStudio
软件工具:
kflash_gui:下载教程,下载地址-github
MaixPy IDE:MaixPy安装教程
参考文档:
MaixPy参考文档
了解MaixPy建议阅读其中的”入门必看指南“;或者,在你遇到问题时可以在改文档中进行搜索,大部分可以找到。
开始干活
在MaixHub训练模型
首先,让我们在浏览器打开MaixHub的网站:MaixHub,任务分为三步:
- 上传数据集
- 创建训练任务,进行训练
- 下载训练好的模型
1、上传数据集
打开MaixHub网站后,点击模型训练
为数据集起个朴素的名字,选择标注类型
- 分类:判断一张图片的类别。
- 检测:比分类更进一步,从图片找到特定物体,得到物体的位置(坐标)和类别
这里我们选择的是分类;检测任务的训练数据标注会比较麻烦。
然后点击进入我们刚刚创建的数据集,现在它里面还什么都没有
接着添加标签,依次输入添加本次任务中所有的标签。
- 我这里是把垃圾分成四类,所以标签有food、harmful、other、recyclable,分别代表厨余垃圾、有害垃圾 、其它垃圾、可回收垃圾。
然后点击一个标签,比如food,接着选择图片,以上传类别为food的图片。
选中所有food类别的图片,然后点击打开,就可以批量一次性上传所有该类别的图片。
图片会要加载小一会儿,加载完成后开始上传。
一定要点击“开始上传”!仅仅加载完是没有用的。
垃圾分类模型训练部署教程,基于MaixHub和MaixPy-k210(2):https://developer.aliyun.com/article/1407167?spm=a2c6h.13148508.setting.33.79f64f0ecKMDuK