在K210开发板上进行人脸识别、口罩识别、手写数字识别等实验后,您可能会想根据自己的需求获得相应的模型。为此,本章将介绍如何使用嘉楠科技提供的在线模型训练工具。
嘉楠科技开发者社区(https://developer.canaan-creative.com/)的模型训练板块旨在简化开发流程,提高开发效率。该平台特别关注视觉场景的落地实现,使用户能够更快捷地完成从数据标注到获得部署包的全过程,并能在搭载嘉楠科技Kendryte®系列AIoT芯片最新一代SOC产品K230的开发板上进行部署。目前,该平台支持图像分类和图像检测两类任务。
要使用嘉楠科技开发者社区模型训练板块,您需要先进行账号注册。进入平台后,您可以开始创建项目。项目创建完成后,在项目列表中可以找到新建的项目。点击项目列表中的新创建项目条目,确定当前选中项目。
接下来,您需要创建数据集。您可以通过上传图片或压缩包的方式添加数据。对于图像分类任务,您需要至少上传两个类别的图片数据。例如,您可以使用鲜花分类数据集进行测试,该数据集可在代码仓库(https://gitcode.com/bin_zhangg0n/K210/tree/main)中找到。请注意,图片数据不能太多,限制在100张左右。
设置好图片和标签后,您可以进入队列进行标注。图像检测任务和图像分类任务的创建方式一致。您同样需要创建一个项目并绑定数据集,然后上传数据并进行标注。
在数据上传和标注完成后,您需要设置训练参数并启动训练。训练过程中会输出日志和损失曲线等信息。训练结束后,您将获得部署资源包,该资源包可以与K230 SDK结合使用,完成后续的开发板开发工作,实现模型部署。
本章介绍了在线模型训练平台的使用,下一章我们将介绍如何部署训练好的模型。