► STM32 model zoo v4.0 (线上)

针对STM32优化的免费AI模型

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

全新4.0版


  • 支持 ST Edge AI Core v3.0
  • 新增 Keras 3 PyTorch 模型
  • 混合精度量化服务和模型
  • 新增应用场景:AFCI、Re‑ID、人脸检测(Face Detection)

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

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

计算机视觉(COMPUTER VISION)
图像分类(IC)24 个模型家族
目标检测(OD)12 个模型家族(包括 ST_YOLO、YOLOv8 和 YOLO11)
姿态估计6 个模型家族(包括 YOLOv8 和 YOLO11)
语义分割1 个模型家族
实例分割 2 个模型家族(包括 YOLOv8 和 YOLO11)
深度估计1 个模型家族
神经风格迁移 – 1 个模型家族
重识别(Re‑Identification)2 个模型家族
人脸检测2 个模型家族

音频(AUDIO)
音频事件检测(AED)3 个模型家族
语音增强 1 个模型家族

时间序列(TIME SERIES)
人体活动识别(HAR)2 个模型家族
手势/手部姿态识别(HPR)1 个模型家族
电弧故障检测(AFCI)2 个模型家族

STM32 Model Zoo 设置要点

要完成 STM32 Model Zoo 的设置,您需要:创建一个 myST 账号,通过 ST Edge AI Developer Cloud 或本地安装方式访问 STM32Cube.AI,确保已安装 Python 3.12.9 版本;如果使用 GPU,还需安装相应的驱动以及 CUDA 和 CUDNN,并在 Windows 环境下避免使用 WSL,以获得最佳的 GPU 加速效果。

STM32现已登陆Hugging Face

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

Model Zoo入门

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