前期准备:亚博K210(或其他K210开发板)

准备数据集

  • 每个标签的图片数量尽量接近
  • 每个标签的图片数量需要大于50张,过少会影响训练效果

进入MaixHub官网创建项目

网址:https://maixhub.com/

首页有详细的在线训练演示

选择模型训练

 创建项目

输入项目名称、项目类别、简介后点击创建

 上传数据集

数据集-->创建数据集

输入名称后点击提交

 

点击确定进入数据集

 

 点击采集数据

 上传数据集 

 选择要上传的数据集

选择后点击上传 

 

 依次上传需要识别的数据后对数据进行打标签

在线打标签

选择整理数据可以看到刚才上传的图像

点击标注数据对数据进行打标签 

 选择添加标签

输入相应的分类后点击添加

按W对需要识别的部分进行框选  按A和D选择上一张或下一张图片

并选择对应的标签

 对所有的图片进行打标签

创建训练任务

  • 批数据量大小:每批训练数据量的大小,对模型效果有一定影响,太大可能会导致内存不足而失败

配置好后点击创建训练任务  

输入任务名称后 点击开始训练

 

 排队等待训练模型

 

 将模型部署到K210开发板上

训练完成后 点击部署

 选择手动部署

 点击下载模型

 将下载的模型进行解压

 将以下文件导入到SD卡

插入到K210开发板上重新上电即可 

修改代码

如果显示屏上显示的颜色反色可以添加以下代码

原代码 

lcd.init(type=1)

添加后的代码

 lcd.init(type=1,invert=1)

识别效果

这里只是进行测试只训练了识别2和3的模型 

可以根据实际情况来训练自己的模型

更多推荐