为了在图像分类中获得满意的结果,通常需要完成费时费力的数据采集和注释任务。 迁移学习技术可以减少添加新类所需的图像数量和训练时间,借此成功应对这一挑战。 在本用例中,我们将这一技术应用于花卉分类,但它还可以扩展到许多其他用例。

方法

STM32 Model Zoo可以找到所需的所有模型,您可以用自己的数据对模型进行训练和再训练
该解决方案使用一种名为“迁移学习”的技术快速训练深度学习模型,来完成图像分类
该模型可以通过STM32 Model Zoo的Python脚本轻松部署在 STM32H747探索套件
演示的用例基于花卉识别。

传感器

视觉:摄像头模块捆绑包(参考: B-CAMS-OMV

数据

数据集 花朵识别数据集(授权许可 CC BY 2.0)
数据格式
5 类花:雏菊、蒲公英、玫瑰、向日葵、郁金香
RGB 彩色图像

结果

模型MobileNetV2 alpha 0.35
输入大小:128x128x3
内存占用:
406.86 KB闪存 用于权重
224.5 KBRAM,用于激活
精确度
浮动模型: 86.78%
量化模型: 86.38%
STM32H747(高频)上的性能 @ 400 MHz
推断时间: 110.27 ms
帧频: 9.0 帧/秒

use-case-stm32-cube-ai-updated-confusion-matrix-flower-recognition use-case-stm32-cube-ai-updated-confusion-matrix-flower-recognition use-case-stm32-cube-ai-updated-confusion-matrix-flower-recognition

混淆矩阵

资源

模型库 ST EDGE AI MODEL ZOO

一系列经过优化的参考AI模型,可在附带相关部署脚本的意法半导体器件上运行。Model Zoo是一种宝贵的资源,可用于为嵌入式应用增添边缘AI功能。

ST EDGE AI MODEL ZOO ST EDGE AI MODEL ZOO ST EDGE AI MODEL ZOO

优化工具 STM32Cube.AI

X-CUBE-AI是一个免费的STM32Cube扩展包,可帮助开发人员将预训练的AI算法(例如神经网络和机器学习模型)自动转换为经过优化的STM32 C代码。

STM32Cube.AI STM32Cube.AI STM32Cube.AI

兼容 STM32H7系列

STM32系列32位微控制器基于Arm Cortex®-M处理器,旨在为MCU用户提供新的开发自由度。它包括一系列产品,集高性能、实时功能、数字信号处理、低功耗/低电压操作、连接性等特性于一身,同时还保持了集成度高和易于开发的特点。

STM32H7系列 STM32H7系列 STM32H7系列
以下案例可能对您也有所启发

工业 | 智慧城市 | 视觉 | STM32Cube.AI | STM32 AI MCU | 合作伙伴 | 视频 | 智能家居

Secure entry systems using id3 face recognition with liveness detection

通过RGB摄像头与ToF传感器进行边缘处理,结合STM32N6 MCU实现快速且安全的反欺骗门禁控制。

工业 | 智慧城市 | 视觉 | STM32Cube.AI | STM32 AI MCU | 合作伙伴 | 视频 | 交通运输 | Model zoo

Smart rear view camera running on batteries

STM32N6 如何提升盲区内人员、汽车、卡车及骑行者的实时检测性能

视觉 | STM32Cube.AI | STM32 AI MCU | 视频 | 客户案例 | 加速度计 | 可穿戴设备

How ultralight AR glasses are redefined with Edge AI

莫界科技利用STM32N6 MCU搭载的边缘AI和计算机视觉技术提升AR眼镜性能