GitHub上用于微控制器的TensorFlow Lite

这是TensorFlow Lite的实验端口,针对微控制器和其他只有千字节内存的设备。它不需要任何操作系统支持,任何标准的C或C ++库或动态内存分配,因此它的设计甚至可以移植到“裸机”系统。核心运行时在Cortex M3上适合16KB,并且有足够的运算符来运行语音关键字检测模型,总共占用22KB。

 项目GitHub网站:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/lite/experimental/micro

如需更多文档,请点击此处:https://www.tensorflow.org/lite/guide/microcontroller

如果想要自定义示例,可以试用此代码实验室:https://g.co/codelabs/sparkfunTF

可以使用Google提供的这个教程训练自己的模型。它拥有一个开放数据集,其中包含 100000 多条志愿者提交的语音,欢迎小伙伴们通过链接帮助扩展此数据集:https://aiyprojects.withgoogle.com/open_speech_recording

 

在STM32上部署TensorFlow Lite Micro需要以下步骤: 1. 准备开发环境:从TensorFlow官方GitHub库中获取TensorFlow Lite Micro的代码,并安装必要的软件工具链和开发环境,如GNU Arm Embedded Toolchain和STM32CubeMX。 2. 配置STM32CubeMX:使用STM32CubeMX配置你的STM32微控制器,包括选择正确的型号、引脚配置和时钟设置等。 3. 构建TensorFlow Lite Micro库:在你的开发环境中,使用Makefile或CMake来构建TensorFlow Lite Micro库。这将生成一个静态库文件(.a文件)。 4. 创建应用程序:根据你的具体应用需求,编写一个C/C++应用程序,包括载入模型、输入数据、运行推理以及处理输出结果等。你可以使用TensorFlow提供的示例代码作为起点。 5. 链接库文件:将TensorFlow Lite Micro库文件与你的应用程序进行链接,以便在编译时正确引用和调用库中的函数和结构。 6. 编译和烧录:使用GNU Arm Embedded Toolchain编译你的应用程序,并将生成的可执行文件烧录到STM32微控制器中。这可以通过ST-Link调试器或其他烧录工具来完成。 7. 测试和调试:在STM32上运行你的应用程序,并通过串口或其他调试接口来观察和验证推理结果。根据需要,进行调试和优化。 请注意,在部署TensorFlow Lite Micro到STM32上时,你需要确保你的STM32微控制器具备足够的计算和存储资源来运行模型。较大的模型可能需要更高级别的STM32系列,而较小的模型则可能适用于低端系列。还需要根据具体情况进行硬件和软件的优化,以确保性能和功耗的平衡。 此外,具体的步骤和细节可能因为使用的STM32型号、开发环境和TensorFlow Lite Micro版本而有所差异。在实际操作中,参考相关的文档、教程和示例代码,并参与相关的开发者社区来获取更详细和具体的指导。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不脱发的程序猿

亲,赏包辣条吧~

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值