目录
前言:
前几天在实验室里翻出了一个k210的板子,正好实验一下训练模型,想要训练一个人脸表情的识别,我去网络上翻了个遍,B站,csdn,github,结果没有找到,于是,我就自己训练了一个模型,适用于最基本的k210.
k210的型号:
在官网中如何选购开发板 - Sipeed Wiki,其中的Maix Dock(M1),Maix Bit,均可以使用这个模型,当然可以将模型放到sd卡中是最方便的,也可以烧录到开发板中。
训练的过程:
主要有几个步骤,分别为:
1:准备数据集。
数据集我在网络中找了好久,有许多的开源,但是都下载不了,但是最后我还是找到了
2:将找到的数据集转化为k210可以识别的大小。
然后使用Image_tool.exe将找到的图片转化为相对应的格式。也就是224的。
3:将处理好的照片使用labeling打标签。
首先建立一个文件夹,如图:
image中是你刚才处理好的图片,xml是在labeling中输出的位置,labels.txt是你要训练的类型有哪些,如图:
因为之前我安装过YOLOv5,训练过模型,所以我直接打开pycharm的终端,然后使用labeling打了大概2000张图片的标签(狗都不打标签,累的不行。)因为网上有许多教我们如何打标签,所以我就不赘述了。比如LabelImg(目标检测标注工具)的安装与使用教程-CSDN博客
4:将打好标签的整个文件夹压缩。
结果如图:
5:登陆到官网开始训练。
训练官网:MaixHub 点开即可。
首页如图:
训练的过程网上也有很多,也不再赘述,比如:K210模型训练小白教程_哔哩哔哩_bilibili
这个up讲的很好。
6.结果:
如图:
最后将训练出的文件双手奉上:scank/biaoq at master (github.com),模型文件在分支master中。
如果想要人脸表情数据集和图片工具的我到时候整理到网盘中,或者再更新到github中,最后求个star。
Thanks♪(・ω・)ノ !!!