NNoM (Neural Network on Microcontroller)
发现一个宝藏嵌入式机器学习深度学习库,可以将神经网络移植到嵌入式设备上,分享给大家。
网站:https://github.com/majianjia/nnom/blob/master/docs/rt-thread_guide.md
NNoM作者:
- 阿嘉 (Jianjia Ma)
1.简介
NNoM是一个专门为了神经网络在 MCU 上运行的框架。 NNoM 是一个定点神经网络库, 现在支持 8-bit定点格式。 当前 NNoM 支持数十种操作,卷积,池化,激活,矩阵计算等等。 此外还提供多种在线评估方法,包括时间统计,内存统计,Top-K 准确度,Confusion Matrix 等等。
它拥有以下优点:
- 快速开发: 电脑上训练的 Keras 模型直接转换成 C 文件。
- 默认纯C后端: 移植和使用畅通无阻,支持32/64bit,MCU/PC。
- 支持复杂结构: 支持多种复杂的网络模型结构。
- 完善的文档: 拥有 API 文档,入门指南,优化指南。
- 入门简单: 多个从简单到复杂的例子,完全开源。
MCU 上的神经网络能做什么?
- 语音关键词识别 (KeyWord Spotting)
- 使用运动传感器识别活动状态 (Human Activity Recognition)
- 神经网络控制系统 (替代PID等传统控制方法)
- 图像处理 (带专用加速器的 MCU)
- ...
** 为什么需要 NNoM **
2014年后的网络,更高效,也更复杂。
CMSIS-NN 之类的库太底层,需要设置的参数众多,不够灵活,只能应用在比较传统的单路径网络上。 而新的网络倾向于从结构上做优化,不可避免地需要在结构上做设计和优化。 复杂的网络结构导致非常难以使用传统的底层库进行部署。 最初,NNoM 在 CMSIS-NN 上封装了一层结构层专门用于处理复杂的网络路径并且简化了参数计算。 后来 NNoM 有了自己的脚本可以一键生成相应的 C 文件,更进一步提升了使用性。 新的纯 C 后端也使得使用 NNoM 后,网络的部署几乎没有平台的限制。
1.1 目录结构
- 在线文档
1.2 许可证
NNoM 使用 Apache License 2.0 许可证,详见LICENSE文件。
1.3 依赖
- RT-Thread 3.0+