Seeed-Studio/ModelAssistant

Seeed SenseCraft Model Assistant is an open-source project focused on embedded AI. 🔥🔥🔥

解决的问题

它简化了将最先进的AI算法部署到计算能力有限、功耗受限的低成本嵌入式设备(如微控制器(MCU)和单板计算机(SBC))上的过程。

工作原理

SSCMA 将现有高性能算法(如 OpenMMLab 提供的算法)针对现实场景和低端硬件进行优化。它提供了一个全面的工具链,包括预训练模型的模型库、基于 Web 的训练与部署工具包,以及专门的框架(SSCMA-Micro)和库(Arduino/Python),弥合模型训练与设备端执行之间的差距。它支持将模型导出为多种格式,包括 TFLite、ONNX、TensorRT 和 UF2,以确保在不同硬件目标上的兼容性。

适用人群

希望将 AI 功能(如计算机视觉或异常检测)集成到嵌入式系统、微控制器(如 ESP32、Arduino)或 SBC(如 Raspberry Pi)中的开发者和创客。

主要亮点

  • 广泛的算法支持:涵盖目标检测(YOLOv8、RTMDet、Swift YOLO)、图像分类、分割、姿态估计和音频分类。
  • 嵌入式优化:专为功耗极低(低于 0.3 瓦)的设备设计。
  • 量化感知训练(QAT):支持 QAT,在减小模型尺寸的同时保持精度,适用于微控制器。
  • 专用场景模型:提供针对特定任务(如模拟与数字电表读数识别)的定制化模型。
  • 端到端工具链:从数据收集、训练到通过 Web 工具包和跨平台框架部署,覆盖完整流程。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目