在STM32上跑神经网络做手势检测
2019-11-211.6万
AI 快速预览详细
本文介绍了在STM32单片机上运行一个不到10层的FCN网络,实现手势检测的方法。尽管硬件性能有限(主频不到100MHz,SRAM不到100K),通过模型量化技术,成功实现了实时手势识别。整个代码包括量化工具都是纯手撸hardcode,没有使用框架。
|
推理一个CNN模型最低需要什么性能的硬件呢? 最近想试试把以前搞嵌入式单片机的一些经验和最近在做的深度学习结合起来,正好前段时间看到新闻说新款iPhone和下一代Pixel似乎都要加入隔空手势识别功能,所以idea就来啦。 为了在Cortex-M的MCU上成功跑起CNN,用的模型是一个不到10层FCN网络,但是即便如此,对于主频只有不到100MHz,SRAM只有不到100K的单片机来说依然是极其吃力的,模型不做量化的话肯定无法做到实时的。 实现细节等有空整理好资料再来详细介绍,先来看看最终效果,整个代码包括量化工具都是纯手撸hardcode,没做成框架,主要是因为,我懒。 |




