► STM32 model zoo v3.2 (线上)

针对STM32优化的免费AI模型

STM32 AI Model Zoo是一个预训练机器学习模型的集合,这些模型经过优化,可在STM32微控制器上运行。 您现在GitHub上找到Model Zoo,如果您想在STM32的项目中添加AI功能,千万不要错过。
全新Model Zoo现已支持STM32N6内嵌的Neural-ART加速器NPU。

全新3.2版


支持 STM32EdgeAI Core v2.2.0
新增了应用场景:风格迁移(StyleTransfer)和快速深度估计(FastDepth)
新增了模型:人脸检测(Face Detection)和人脸关键点(Face Landmarks)
架构和代码库清理优化

获取适合您应用的最佳边缘AI模型

– 提供大量面向应用的可重训练模型。
– 脚本支持轻松重训练用户数据集中的任何模型。
– 根据用户AI模型自动生成应用代码示例。

图像分类 (IC):模型:EfficientNet、MobileNet v1、MobileNet v2、Resnet v1(包括混合量化)、SqueezeNet v1.1、STMNIST。

目标检测 (OD):模型:ST SSD MobileNet v1、Tiny YOLO v2、SSD MobileNet v2 fpn lite、ST Yolo LC v1。

人类活动识别 (HAR):模型:不同设置下的CNN IGN和CNN GMP。

姿态估计:模型包括 YOLOv8n、MoveNet 和 Hand landmarks(手部关键点)。

实例分割: 模型包括 YOLOv8n。

语义分割: 模型包括 Deeplabv3。

手势识别 (HPR):模型:ST CNN 2D Hand Posture。

音频事件检测(AED): 模型包括 Yamnet、MiniResnet、MiniResnet v2。

语音增强: 模型包括 STFT-TCNN 音频降噪。

STM32 Model Zoo设置要领

要设置STM32 Model Zoo,您需要创建一个myST账户;通过ST Edge AI Developer Cloud本地安装访问STM32Cube.AI;确保已安装Python 3.9至3.10.x版;如果使用GPU,请安装相应的驱动程序以及CUDA和CUDNN,避免在Windows上安装WSL,以获得最佳的GPU加速效果。

STM32现已登陆Hugging Face

所有STM32 Model Zoo服务均可通过Docker使用。
适用模型卡浏览NN模型性能。

Model Zoo入门

探索如何优化AI神经网络以及为STM32项目创建处理库