模型文件通常包含了神经网络架构的参数和权重信息,是神经网络训练过程的结果。对于K210而言,模型可以用于人脸检测、口罩识别、人脸识别、物体识别等各种机器学习任务。由于K210芯片资源有限,所以训练好的模型需要传输至TF卡储存,模型文件的后缀为.kmodel
传输模型文件至TF卡的方式有两种,方式一是使用读卡器,将TF卡插入读卡器,读卡器插入电脑USB口,下载资料中的模型文件压缩包并解压,将KPU文件夹复制到H:盘的根目录。
方式一可以一次导入所有模型文件,省去每次导入模型文件的操作,以不同路径来区分加载模型文件。所有模型文件都放到KPU文件夹下,请不要修改KPU以及KPU目录以下的文件名称,否则默认例程程序会出现找不到模型文件的问题。
方式二是通过CanMV IDE来导入模型文件,无需借助读卡器,这里以导入landmark68.kmodel为例,将K210模块通过microUSB数据线连接到电脑USB口。然后点击CanMV IDE左下角连接按钮,连接成功会将图标变为已连接状态,同时灰色播放键变为绿色。
打开工具->保存文件到CanMV Cam。
点击打开,然后选择要传入的模型文件,传入内存卡。
注意:这里写入的文件优先写入到内存卡,保存时请删掉文件路径,带路径保存容易报错。
运行代码时,以具体保存的模型路径为准,如按照本教程方式一操作,则使用默认代码,无需修改。
如按照本教程的方式二操作,需要将程序中的多余路径删掉,即仅保留前缀/sd/。