Vision Board系列教程 | 神经网络模型训练及部署指南

2024-10-17

准备工作

win10/11系统的电脑

建议1张FAT32格式的SD卡

建议自备1根Type-C数据线



在正式进行开发前,需要安装下图中勾选的四个软件!!!

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第1张

注册账号并创建Edge Impulse工程

打开Edge Impluse网站,注册登录。随后在项目标签页创建新项目:

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第2张

上传训练集

按照以下步骤,依次点击Dashboard->Add existing data->Upload data。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第3张
Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第4张

注意:

若未在数据集中对图片进行label注释(可参考官方文章),需要手动进行添加标签,或直接选择Enter label进行标签定义。分3次单独上传3个不同图片的文件夹,并添加label为ship、truck和plane。

生成特征

点击左侧Impulse design->Create Impulse,依次点击Add a processing block、Add a learning block进行输入数据对象、训练模型的选择,并点击保存。(Add 带星标的即可)

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第5张

继续点击左侧Image,Color depth选择RGB,点击保存。接下来会自动跳转到生成特征界面,点击生成特征按钮,等待特征生成,结束后会有三维图像显示。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第6张

迁移训练

点击左侧Transfer learning按钮,依次设置训练参数:训练周期、学习率等参数。随后选择自己实验最符合的训练模型(默认选第一个),点击开始训练。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第7张

如果最后的结果和准确率不满足自己的实验要求,可尝试重新训练,重新更改参数及训练模型。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第8张

在Vision Board上部署

在左侧点击Deployment,搜索OpenMV library,点击Build,等待固件生成。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第9张

将下载下来的压缩文件解压("trained.tflite"、"labels.txt"、"ei_image_classification.py"),将ei_image_classification.py改名为main.py,随后将3个文件全部复制到sd卡中去(取下SD卡,使用读卡器将3个文件复制到sd卡)。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第10张

用Type-C线连接Vision Board USB-OTG口,随后将sd卡中的main.py拖入OpenMV IDE中,打开并运行,此时可以在串口终端看见识别的结果及准确率。

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第11张

使用虚拟U盘功能

点击RT-Thread Setting-->TinyUSB->配置项

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第12张

在TinyUSB选项处依次点开USING USB device->Using Mass Storage Class(MSC),将The name of the block device used by MSC中的内容更改为sd,随后保存。重新点击编译,编译完成后即可烧录到开发板。(由于文件较大,时间可能会比较长)

注意:烧录需要将Type-C线插到USB-DBG口,烧录完成后再将线插到USB-OTG口,使能USB复合设备后,每次第一次上电需要先等待弹出U盘后再连接 OpenMV IDE!!否则会卡死!!

Vision Board系列教程 | 神经网络模型训练及部署指南 (https://ic.work/) 技术资料 第13张

弹出U盘后就可以把生成OpenMV固件拖至u盘中,随后连接OpenMV IDE,即可体验自己训练的神经网络模型。





文章推荐

相关推荐