1.拍摄照片,每一种模型尽可能从多个角度拍摄,这样识别率会比较高,照片的数量在20张以上会比较好。
2.修改照片格式,这里用到一个工具Image_tool来帮助大家,工具放在下面了。
先在桌面创建一个文件夹,取名num(你也可以随便取)
再在这个文件夹里面创建一个文件夹,取名images(这个要和我的一致)
打开Image_tool这个工具,点击选择图像文件夹
选择刚才我们创建的那个文件夹,选择images这个文件夹
然后工具就开始帮我们修改图片格式(这里做演示,我的图片就比较少了)
最后在我们的num这个文件夹里会出现名为images_out的文件夹
我们把之前的image文件夹删掉,把image_out文件夹名改为images
3.标注训练集
在num这个文件夹里新建一个名为xml的文件夹,新建一labels的文本文档
打开名为labelImg的一个工具,在里面可以给训练集做标记
点击open Dir,这个是要选择图片的文件夹
选择images这个文件夹
点击Save Dir,这个是我们要保存的文件夹
选择xml这个文件夹
点击View,给第一个Aout Saving勾上,这个是自动保存的设置,这里有几个快捷指令给大家,点击W就可以给图片框选标注的区域,标签自己定义,A是上一张,D是下一张(注意是英文输入法才可以用哦)
4.网站在线创建项目,打开嘉楠开发者社区网站https://developer.canaan-creative.com/index.html?channel=developer#/model/records
需要自己先注册一个账号,然后登录
登录后选择模型训练,创建项目
项目名称自己定义,项目类型选择图像检测
提交后在我们的项目列表里就有刚刚创建的项目了,选择刚刚创建的项目
确定选中项目
这时我们的项目就选上了,颜色加深了
点击数据集,创建数据集
给数据集取名,提交
绑定数据集,提交
点击采集数据,选择压缩包,这里要先把刚刚我们的num文件夹压缩一下
点击开始上传
如果有遗漏标注的,或者打不开标注工具的可以点击左侧的标注数据,重新标注
点击左侧训练记录
创建任务,命名
到这里训练的操作过程就结束了,等待训练结果出来
5.开始训练,等待一段时间,训练完成会把相应的模型通过压缩包的形式发送到你注册时的邮箱
6.打开压缩包,把模型文件拷到SD卡里
7.把代码输入到k210开发板里,这里需要注意的是需要修改代码里类的名称,和相关的训练数据。观察是否达到自己想要的效果,如果不满意就重新训练,直到自己满意为止
最后希望对大家有所帮助,谢谢
工具包链接: