STM32上小白也能快速部署机器学习模型工具——NanoEdgeAIStudio初使用体会
前几日,偶然看见ST官方推了一个视频,介绍其边缘计算模型训练软件NanoEdge AI Studio,并且还是免费使用。
而我在嵌入式边缘计算部署中卡住的则是模型训练,获取和优化的步骤。因此这个软件如果可以快速训练部署的话实在是可以帮我的大忙。
于是连夜赶忙去官网查阅其相关资料和下载软件。

从官网(需要注册)下载和安装好NanoEdge AI Stuido之后,打开软件需要进行激活。
在其官方文档中可以找到相关的离线激活教程,下载软件的时候需要填写申请邮箱,之后邮箱会收到一份激活邮件。
PS:(这里千万不要使用离线激活,哎呀呀折腾死我了)
将这串密钥填入NanoEdge中点击激活,会得到一串新的License key
软件界面分为了以上五个主要部分:
- 创建项目
- 打开已有项目
- 帮助文档
- 开源数据集(这里不知道是不是我是离线的原因)
- 工具栏
测试NanoEdge
这里我尝试这创建项目,用一个2分类模型(n-class中n=2),我们简单的来测试一下,简单的做一个信号识别吧,虽然很没有必要,但是主要起到演示效果。

在项目配置中,除了传感器类型,基本没啥好讲述的。
在传感器类型中,有许多选项给我们选择,主要是数据源的由来,Generic可以作为通用传感器输入,其轴数(Number of axes)由数据源决定,例如我们有一个三轴姿态传感器,那么其xyz轴就是数据就是三轴,我们先简单的设置一个一轴来使用。后面几个就不一一赘述了。
选择芯片,这里支持大部分STM32系列。
我们选择自己常用的芯片,这边我选择F4的,这里我居然看见了Arduino,Nano和Uno用户的福音了。
在这里导入数据,文档说明支持TXT和CSV格式(不知道为什么机器学习的文件格式都避开xls)
文档中具体说明了由三种导入方式,可能我们日常用的话会比较从串口导入吧(拜托看看DMA+ADC采样然后配合串口打印,真用上了,对ADC数据的模型构建,完了写这里越写越激动,感觉好多好多东西想去做,谁懂半夜两点钟写文章越写越清醒)

用C语言简单的写一个正弦信号脚本。
导入脚本,选择分割方式,选择预览数量。
这里由于二分类问题,我们需要给每个类情况都进行导入(分类1,分类2)

导入两路信号,一路正常信号,一路载波信号。 需要注意的是:首先信号数量要是2的倍数,第二信号上要有噪声,不能是太完美的信号,第三如果相邻的数据相同可能会被识别为过采样,因此需要设定好数据。
全是勾勾就没有问题啦。
OK,咱们开始跑模型,这里实际使用的情况应该需要多组数据。
之后就是等待训练结束。

训练结束后,其模型大小优化后Flash 4.3K(STM32F407ZGT6的FLASH有1M)完全存放的下这个模型。
但是由于我的数据集问题,具体要看使用的训练集。

之后编译我们的模型。
保存我们训练好的模型。

之后几天出具体使用这个模型的示例!
使用体会
总的使用下来来说,不需要什么过硬的专业技能,在数据除了最开始因为离线激活导致后面部署出了点问题之外几乎没有什么卡住我的地方(数据集的地方是没想到数据集太过于理想但是实际情况下肯定是没有我这么理想的)所以第一次使用大部分的时间都花在了官网的参考文献上了


登录 或 注册 后才可以进行评论哦!
还没有评论,抢个沙发!