STM32 AI Model Zoo 项目推荐

STM32 AI Model Zoo 项目推荐

stm32ai-modelzoo AI Model Zoo for STM32 devices 项目地址: https://gitcode.com/gh_mirrors/st/stm32ai-modelzoo

项目基础介绍和主要编程语言

STM32 AI Model Zoo 是由 STMicroelectronics 开发的开源项目，旨在为 STM32 微控制器提供优化的机器学习模型。该项目主要使用 Python 作为编程语言，同时也涉及 C/C++ 用于嵌入式系统的部署。

项目核心功能

STM32 AI Model Zoo 的核心功能包括：

1. 模型优化：提供了一系列经过优化的机器学习模型，这些模型专门针对 STM32 微控制器进行了优化，以确保在资源受限的环境中高效运行。
2. 模型训练与再训练：支持用户使用自己的数据集对模型进行再训练，或者从头开始训练新模型。
3. 应用代码生成：自动生成与用户 AI 模型相对应的应用代码，简化部署过程。
4. 性能评估：提供对模型在 STM32 设备上的性能评估，包括浮点和量化模型的性能。

项目最近更新的功能

最近更新的功能包括：

1. 版本 2.0：

   • 统一和规范了所有用例的架构。
   • 引入了模块化设计，支持不同的操作模式（训练、基准测试、评估、部署、量化）独立运行或链式运行。
   • 通过 YAML 配置文件简化代码入口点，配置所有所需服务。
   • 支持 Bring Your Own Model (BYOM) 功能，允许用户重新训练自己的模型。
   • 支持 Bring Your Own Data (BYOD) 功能，允许用户使用自己的数据集对预训练模型进行微调。

2. 版本 2.1：

   • 增加了与 STM32MP257F-EV1 板兼容的额外模型。
   • 支持逐张量量化。
   • 集成 ONNX 模型量化和评估支持。
   • 支持 STEdgeAI（STM32Cube.AI v9.1.0 及后续版本）。
   • 扩展用例支持，包括姿态估计和语义分割。
   • 标准化日志信息，提供统一的体验。

stm32ai-modelzoo AI Model Zoo for STM32 devices 项目地址: https://gitcode.com/gh_mirrors/st/stm32ai-modelzoo