【RK2206-小凌派】使用TinyMaix轻量级AI推理框架识别手写数字

LJH1095 2024-07-01 12:01:02 阅读 82

一、TinyMaix 介绍

TinyMaix是国内sipeed团队开发一个面向MCU单片机的轻量级AI推理框架,以下是官方介绍:

TinyMaix 是面向单片机的超轻量级的神经网络推理库,即 TinyML 推理库,可以让你在任意单片机上运行轻量级深度学习模型。

关键特性

核心代码少于 400行(tm_layers.c+tm_model.c+arch_cpu.h), 代码段(.text)少于3KB低内存消耗,甚至 Arduino ATmega328 (32KB Flash, 2KB Ram) 都能基于 TinyMaix 跑 mnist(手写数字识别)支持 INT8/FP32/FP16 模型,实验性地支持 FP8 模型,支持 keras h5 或 tflite 模型转换支持多种芯片架构的专用指令优化: ARM SIMD/NEON/MVEI,RV32P, RV64V友好的用户接口,只需要 load/run 模型~支持全静态的内存配置(无需 malloc )即将支持 MaixHub 在线模型训练

1.1TinyMaix可以简单理解为一个矩阵和向量计算库,目前已支持如下几种计算硬件: 

TinyMaix官方代码仓:https://github.com/sipeed/tinymaix

#define TM_ARCH_CPU         (0) //default, pure cpu compute

#define TM_ARCH_ARM_SIMD    (1) //ARM Cortex M4/M7, etc.

#define TM_ARCH_ARM_NEON    (2) //ARM Cortex A7, etc.

#define TM_ARCH_ARM_MVEI    (3) //ARMv8.1: M55, etc.

#define TM_ARCH_RV32P       (4) //T-head E907, etc.

#define TM_ARCH_RV64V       (5) //T-head C906,C910, etc.

#define TM_ARCH_CSKYV2      (6) //cskyv2 with dsp core

#define TM_ARCH_X86_SSE2    (7) //x86 sse2

二、 RK2206-小凌派介绍

小凌派-RK2206是一款专为HarmonyOS系统打造的HarmonyOS开发板,以下是官方介绍:

小凌派-RK2206是一款由福州市凌睿智捷电子有限公司专为HarmonyOS系统打造的HarmonyOS开发板。

采用IoTCluB的E53接口标准,可接入所有的E53案例扩展板。主控为瑞芯微RK2206芯片,200MHz主频,RAM 256KB、ROM 16KB、PSRAM 8M、FLASH 8M。板载NFC Forum Type 2 Tag芯片及天线,可实现HarmonyOS“碰一碰”功能。支持 IEEE 802.11b/g/n 制式,AP 模式。USB 2.0 、SD card /SPI/I2C 接口、麦克风接口、E53 接口(包含 UART、 I2C、ADC、DAC、PWM、5 个 GPIO)。一根TypeC USB线,即可实现供电、下载等功能。一根TypeC USB线,可支持串口调试功能。

小凌派-RK2206官方代码仓:小凌派-RK2206开发板: 小凌派-RK2206开发板鸿蒙3.0稳定版资料共享

三、在HarmonyOS源码的lockzhiner-rk2206-openharmony3.0lts/vendor/lockzhiner/rk2206/samples路径下添加 tinymaix-mnist 文件夹,并从TinyMaix代码仓库中复制以下文件到对应文件夹下。

3.1 在arch_arm_simd.h中添加宏定义。

#define __UVISION_VERSION 1 

3.2 在tm_port.h 中修改TM_ARCH的宏定义。

3.3 在tm_port.h 中修改TM_OPT_LEVEl的宏定义。

3.4  在TinyMaix_mnist文件夹中添加BUILD.gn。

3.5 在sample的BUILD.gn 文件中添加

"./tinyMaix-mnist:tinymaix_mnist",

 3.6 使用hb build -f 编译工程

3.7 使用MobaXterm终端软件,查看RK2206的串口数据,可以看到串口打印出预测的数字 2。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。