一、cube.AI简介及cubeIDE集成2 F0 G7 s. t- G- L: a* E$ I6 M
1.1 cube.AI介绍
6 |, N, u g- E. ` cube.AI准确来说是STM32Cube.AI,它是ST公司的打造的STM32Cube生态体系的扩展包X-CUBE-AI,专用于帮助开发者实现人工智能开发。确切地说,是将基于各种人工智能开发框架训练出来的算法模型,统一转换为c语言支持的源码模型,然后将c模型与STM32的硬件产品结合,实现人工智能模型可直接部署在前端或边缘端设备,实现人工智能就地计算。
- o* q# P6 C; |3 y1 n
; [; g$ K: F5 F! k/ l cube.AI 以插件形式支持ST相关开发平台如cubeIDE、cubeMX、keil等,整体开发过程分为三个主要部分,1)收集及治理数据,2)训练及验证模型,3)c模型生成及前端或边缘端部署,如下图所示:
+ C. ]8 T [6 T4 [4 @
/ s" q w: X. B, G5 K3 }' L' ?. v
2 J# L8 l9 ]8 g4 U) E
- B. R2 H3 s4 C0 g, X* h
目前cube.AI支持各种深度学习框架的训练模型,如Keras和TensorFlow™ Lite,并支持可导出为ONNX标准格式的所有框架,如PyTorch™、Microsoft® Cognitive Toolkit、MATLAB®等,然后通过 cube.MX可视化配置界面导入这些深度学习框架导出的训练模型来配置及生成c模型,进而部署在STM32芯片上。
/ f6 b- V7 H. n/ ^( V3 U' j
1 t( O: Q) q: K- u/ {3 U: h
% r- P: C% v- t6 S4 }8 U( x" ^" c 1.2 cube.AI与cubeIDE集成与安装 D( T: V) ]8 q6 P, q
在cubeIDE的帮助菜单栏,选择嵌入式软件包管理项(Embedded Software Packages Manager)进入X-CUBE-AI扩展包安装页面。选择X-CUBE-AI需要的版本进行安装即可,如下图所示,安装完成后,该版本前面方框呈绿色标记。
8 v- y( E2 T2 } ^% ^
; N( r/ ?$ E- W
" A! x8 m6 t& b+ ]
: t$ K. V; g6 u4 s' k2 S- Z- v
1 c) M, P0 q5 K% U9 s 1.3 cube.AI支持硬件平台! r _+ P3 H4 I- A4 P
得益于ST公司不断的优化及迭代X-CUBE-AI扩展包,神经网络模型生成c模型后得以使用更小的算力资源和几乎无损的算法精度,因此使其能部署到STM32绝大多数的芯片上,目前其支持的MCU及MPU型号如下图所示。! x9 Y6 o9 E* ?& O) H
( i6 {! @4 ]1 I. l$ m/ v( y' J
. F5 x( b+ b! W* b& s
; \8 X% M2 @5 N0 B- i: g: { 1.4 cube.AI应用的好处, t1 K7 J+ B2 X/ {- c
将神经网络边缘化部署后,减少延迟、节约能源、提高云利用率,并通过大限度地减少互联网上的数据交换来保护隐私,而结合X-CUBE-AI使得神经网络部署在边缘端端的便捷、灵活、低成本,微机智能成为更多产品的选择。
* h$ T; \" g4 t. l0 ^0 x
7 a) ?- I; {$ W, Q, Y$ a
3 b3 c/ K `1 B, ^2 `' y' J
* E/ B5 I, ]9 T7 u l# }9 ?- L 二、FP-AI-SENSING13 s5 N, x: Q4 j/ V& F1 P
2.1 FP-AI-SENSING1简介
: I" [ m+ k/ Z, h3 [- j FP-AI-SENSING1是ST公司提供的STM32Cube.AI示例,可通过BLE(低功耗蓝牙)将物联网节点连接到智能手机,并使用STBLESensor应用程序,配置设备,实现数据采集,使得用于训练神经网络模型的数据更贴近实际使用场景,具有更好的训练效果和精度。! p7 Q# z! Z+ z2 _1 B3 w
1 k& [7 h, ]; i# x& u% N; h 在FP-AI-SENSING1案例页面,下载源码包及其数据手册。
& |9 D8 d: B+ l& ?% t: H3 U$ d/ ~ z4 ]3 N
' p" P1 a3 z2 ~7 B; ]- g6 x
: s) j6 Y+ Z2 j4 x3 K
2.2 FP-AI-SENSING1软件包支持硬件平台
# T8 Y6 b% [- E' _- z3 V ST公司为FP-AI-SENSING1示例运行提供了硬件平台,支持开发者快速学习了解FP-AI-SENSING1示例,从而了解Cube.AI的开发过程。# A3 \5 D9 s) Q F* d
2 E1 }+ t: c1 M' F
& O7 I5 h% e% [3 m# a3 |; P8 {3 {# Y# {; A+ F9 `; W3 l1 ?+ s% F
三、FP-AI-SENSING1部署
4 V% F( t: }( |6 T" [$ m 3.1 B-L475E-IOT01A开发板" W7 V8 Z5 I# j L
本文采用ST公司的B-L475E-IOT01A开发板,打开CubeMX工具,选择Start my project from ST Board,搜索B-L475E-IOT01A,如下图所示,可以在1、2、3页面下载开发板相关的原理框图、文档及案例、说明手册。
: H4 v- \% M- v! D, ^
. X0 a2 z& W2 b& P
/ y7 u7 [7 V' P4 |: q/ I# A: H5 C* ?. D$ T
3.2 FP-AI-SENSING1软件包下载及配置" c3 K B$ a. K H* K
下载FP-AI-SENSING1软件包后,解压下载的源码包:en.fp-ai-sensing1.zip,进入“STM32CubeFunctionPack_SENSING1_V4.0.3\Projects\B-L475E-IOT01A\Applications\SENSING1\STM32CubeIDE”目录,用文本编辑工具打开“CleanSENSING1.bat”,(linux系统的,采用CleanSENSING1.sh文件)。/ R8 t5 A3 \3 k# k6 a
f0 s3 B' a( C) W) E
2 M# q ~( R+ B% Q) o0 e: x
9 p: i: d) u; q* f4 c CleanSENSING1.bat运行依赖STM32cube生态的另一个开发工具:STM32CubeProgrammer,该工具可以帮助开发者读取、写入和验证设备内存等。
- Z$ Z8 d: W- ~: p3 u6 e, d' G% A4 r# ^8 \- B' t
在STM32CubeProgrammer工具下载页面,下载该工具及说明手册:
1 O/ d& j* b2 {, U7 w% U c) g4 A. W& P/ O; s; k4 j) \
* ?; ~7 ^& d/ O& \- Y 下载并安装STM32CubeProgrammer工具,例如本文安装目录为:D:\workForSoftware\STM32CubeProgrammer
. {& z- W2 u& K! x" p
6 N: f& K. I; M# ]# l) i" S% s
/ |1 I6 A. Q3 i5 y+ M! ~" n- q
& N- H/ N- `* M( @9 P 修改CleanSENSING1.bat依赖工具“STM32CubeProgrammer”的路径:
5 G$ o5 f) D0 v. J! E, W/ ~
/ i+ A3 L& S0 Y; a, w! D
( S7 Y2 f+ w( {- d, }. ?' [8 P6 B
( d+ _* x, K1 r! [: p3 G) w4 l 3.3 固件烧录& E$ w, F6 {) [" ]4 {* m1 ]/ t2 d
将B-L475E-IOT01A开发板用Micro USB连接到电脑上,/ ~; j/ r/ m/ W4 _. \* B4 E, A( z
- f# _* \6 q4 _* _+ p1 n
' ]. @' x* O5 J: H1 Q: ]! a. V& b; ^1 p, ^9 C
连接之后,驱动会自动安装,进入设备管理页面,确认串口编号和配置串口参数。% q: N, ]. ~% ]( x. r" O# b( M
; v2 Z5 x3 r7 |. e8 S2 ~, @
0 f! J0 b. @, C9 W1 y4 V
E6 I. {& ~& k; R# ~) l( K 右键CleanSENSING1.bat文件以管理员身份运行,将在开发板安装引导加载程序和更新固件。1 F; ], x: y! a: I
) i( Q. N! F# N3 U3 Z
: @! z, J; \. e* @; a
$ O4 e% \/ w9 b8 \# l- }6 G7 c+ d
该脚本可以对B-L475E-IOT01A开发板实现以下操作,
! J+ Q) _- S% C; k: w) t) X•完全闪存擦除
; X- u: e( ?! u•在右侧闪存区域加载BootLoader0 e' G( ?$ N& {
•在右侧闪存区域加载程序(编译后)
- M! @6 j) W. T0 c5 X( }, t2 N( r6 M: [•重置电路板
0 i% L+ T# Z8 \! {5 y& F( C$ x3 I/ B8 }4 h8 q z
; M3 j0 i/ e) u/ W. m+ p3 ]( L$ W% u" | 3.4 FP-AI-SENSING1示例工程部署
' {( E% C0 l8 A' R' K" u% w0 m& ^4 r 在该目录下,进入“B-L475E-IOT01A”目录,用CubeIDE打开.project,打开FP-AI-SENSING1工程。
. W8 L) o" G! z) _. k. e8 O( ]2 b
, n. H0 R' \- X# E. R2 K
: v6 F; P3 G6 d
, ~" @' [; a7 M* T1 q1 E 打开该工程后如下图所示,用户可调整源码在User目录,关于本工程信息请查看readme.txt文件。
' u# K0 n) X) f) X1 [3 g- ^
# W2 O; q8 U+ }3 S( A
" ^+ A+ C' r" N3 \- C! I6 a' d5 ?+ I8 h3 {% W/ W$ T
在main.c函数中找到Init_BlueNRG_Stack函数,该函数可以设置BLE(低功耗蓝牙)的服务名,9 q7 ~4 B8 ~* _8 b
- static void Init_BlueNRG_Stack(void)
3 i5 t. ~) Z I- d8 u9 G& [ - {
/ a, s- `+ O; V$ D2 r - char BoardName[8];% k6 l" O. F/ L9 D" j0 l
- uint16_t service_handle, dev_name_char_handle, appearance_char_handle;& d+ T. B8 E' C: {
- int ret;& H# D6 S( N* f G
-
- V5 V1 a4 \ A - for(int i=0; i<7; i++) {% |& m( e1 _ p* ~' @
- BoardName[i]= NodeName[i+1];
3 U. P2 z, {" Z7 j - }
复制代码
3 g0 z9 P. @' F' W2 C! H+ S6 t 该函数采用默认的BLE名称,该默认名称定义在SENSING1.h设置,例如:IAI_403- {) D2 z9 Q$ {) K' A) {( O" w0 L. W) V
$ |1 m% u- l7 U# _
# w! \7 k6 i! P% L! _& N, H/ T Y: t7 I8 b3 ^
现在调整BLE名称为AI_Test& o2 Z: I+ I+ f2 g1 Z
- static void Init_BlueNRG_Stack(void)
, f% {( Y9 R( u6 D: N* s$ I, v0 D1 l" b. s - {
, `7 [ L8 d8 j& T; x - // char BoardName[8];/ `% |- L2 @' X3 x
- char BoardName[8] = {'A','I','_','T','e','s','t'};+ a- \& N$ {& q- m0 |$ H* D
- uint16_t service_handle, dev_name_char_handle, appearance_char_handle;
P! Z2 \/ J# V3 y - int ret;
* w+ L+ |' @% f' N+ j# r - & O- s/ X0 v% N$ y2 g) o% w
- for(int i=0; i<7; i++) {2 c# ~8 w$ V$ m- Q/ U
- // BoardName[i]= NodeName[i+1];
8 V q( u+ O0 P( J3 s - NodeName[i+1] = BoardName[i];
* e8 v5 T' d: p - }
复制代码
! O( F+ _% b% j6 x$ x. I {, t 配置工程输出格式支持如下:
1 g! s/ E: g& {" U) c7 v9 i" {9 m, @- G! N
, S5 R% x9 b/ E( U* a
; f8 [! D% A" s6 r3 X/ Z6 `2 T 配置运行设置如下:
* {% b6 {/ x# h. u! Z/ i L; k/ @4 E, m* Q/ P: R: P1 K5 q
4 ?9 C' N. `* G, s
2 H! x O" I5 k. T1 h. {* a4 f 然后编译及下载程序:% e5 s/ x3 {7 E2 v
5 u. l5 f+ T( d6 N- q
5 o7 i6 T2 F2 ?. X/ t4 c
6 y3 i) { r1 r2 f 打开串口工具,连接上对于串口,点击开发板上的重置按钮(黑色按钮),串口日志输出如下,日志显示BLE模块启动成功:% C" p; P2 U% G4 u9 e
; t( C0 F7 N4 W
. U+ O6 N7 N) @. P. N5 W% c7 D
( i& \& m" ~" K- S. h
7 Z1 w7 a: Z* ?0 Z; ^ 四、数据采集
4 |. Y3 Z9 z2 | 4.1 STBLE-Sensor软件下载安装; B; i7 p$ A9 D6 P3 |7 d/ d0 b
确保手机支持低功耗蓝牙通信,进入ST的BLE传感器应用下载页面,4 z+ a6 h9 o; L+ Z& O( B
+ Y) E$ N. H. J0 L' W1 N- ] 下载对应的手机应用程序:4 z1 Q, t) s" y3 {
! n$ o5 N1 v/ w3 r& r& Z- c
' [6 s3 [ K9 w& z4 ]
$ k* D2 C% [. | p, G3 L1 {$ _
$ P$ `! {& D* y- d0 ]. w6 S 4.2 STBLESensor配置数据采集
+ \4 _5 y+ s# n/ n 本文用的是华为手机及android系统,安装完成APP后启动进入界面(当前版本是4.14),点击搜索,得到AI_Test蓝牙服务名。" X X9 K! t3 x3 i
! Q% ^9 [5 w6 ?; W& g9 Y
7 a, z% e$ T: q1 C& i- i
8 y% n) F; s. ^" i0 V8 i C
选择AI_Test蓝牙服务后,进入页面,(android)在左上角菜单下拉菜单选择,Data Log(sensing1),进入数据采集页面,选择Accelerometer(三轴加速度计),并设置参数为1.0Hz、26Hz、1.0X。* c4 [$ |0 [* d4 [
& B* T9 h& \& y! P+ V. g' J
& I$ ^1 ^' T6 P/ k8 |7 m1 U$ ?- C
3 s H+ x7 z. R7 l' v
9 J. ^! O4 z$ U4 B0 e3 A+ P: j$ r 在数据记录操作页面,先新建标签,例如Jogging(慢跑),Walking(走了),Stationary(静立)等等。
5 x: x" |, X( K4 K 1)开启数据采集记录时:先打开标签,再点击START LOGGING按钮开启 r3 V$ {8 n% _4 T( g3 A
2)关闭数据采集记录时,先点击START LOGGING按钮关闭,再关闭标签。
: k( J5 {9 \: Y7 g+ U: n' `1 _( @! j+ y9 Y6 ] ^, l8 ~% A; p
1 D/ Y7 a6 Y8 Q9 a
I0 P/ m) h- R6 J4 @
例如,本文按上述操作记录了Walking、Jogging两次数据记录,将生成两个.csv文件。
; g3 H3 g, e1 x) Y( Q
@5 B4 v8 |) S7 {! V2 `3 B! t- _5 ~. `3 @: S" j
五、数据治理及模型训练* \$ A! C! |7 s s# R3 V6 o
5.1 从开发板取出采集记录数据文件
1 j$ M: |- W1 v5 i 断掉开发板与电脑的USB连接,在开发板背面将在1-2跳线帽拔掉,插入5-6跳线,然后USB连接从ST-LINK连接转到USB-OTG接口接线,如下图(1->2)。
f$ a* t3 A' X& ~
1 [! z+ U9 p( r3 D9 P
) D: {$ E/ N* b! V5 o, C/ X
' C3 X! y$ Y2 m6 H* y( W q* t
开发板重新上电后,保持按下user按钮(蓝色),同时按下reset按钮(黑色),然后先松开reset按钮,在松开user按钮,激活USB-OTG。
1 x; T2 j1 ^, @- h+ v
, h1 A: k. E# u) R
6 K' e3 A0 r: a7 C ]4 ~5 V' s: B: l( C3 D K V& {0 o V
USB-OTG激活后,开发板将作为一个U盘显示在电脑上,里面有刚才数据采集保存的CSV文件。
" X; E7 }/ a* K8 N# z7 z: S
! c! f. \* s& ]6 a! W# E
: f' P% e! k7 I% g
& q* J# z; _& H
1 V; o4 M: a; ]6 z: e8 O0 } N7 Y2 \
在“STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR”目录创建一个文件目录Log_data,将该文件拷贝到该目录下:" y4 E' z8 d, j9 G2 M
5 C2 G8 G4 R5 e/ e
- ^* b7 i0 b* K& ~
* s9 k# S7 V: `: `* b
7 O- }8 V% E3 U$ R- g# C
9 {) s6 e) z$ E8 M7 \ 该CSV记录数据格式如下,时间、行为、三个传感数值:: }/ e& s F' o6 O' c( s
" H6 w g& {3 J9 k/ V) ~' Y
& V1 _6 Z$ T6 r. {
1 `' h0 _1 G' f; r6 C
& o- T9 ^# N( d6 b5 k! Y 5.2 神经网络模型训练
9 A! v2 R% j* a, R& U1 o$ e “Training Scripts\HAR ”是官方提供的一个人类行为姿态识别训练项目,默认是采用,采用Keras前端+tensorflow后端实现。先安装tensorflow、Keras等支持。0 G/ u* Q x, t& [
7 V2 S/ h9 b7 y9 E! d8 i
本文安装如下:; X n1 E7 x q4 Y$ i. P
- #已安装python3.6
% d# K: W. l, {/ u7 I+ A - pip3 install tensorflow==1.14 -i 4 }: s# ?! e+ _ r4 l( N; O
- 6 }6 H0 T7 l& A& e6 _& G
- ERROR: tensorboard 1.14.0 has requirement setuptools>=41.0.0, but you'll have setuptools 28.8.0 which is incompatible.
& K% o7 N+ e) D& R! T/ Q! M - python3 -m pip install --upgrade pip -i
3 l0 l% F5 h7 C' ?" E( O2 t - pip3 install keras==2.2.4 -i
复制代码
$ e2 C$ z, ?' M2 [% x( q! q 根据HAR 项目的readme.txt通过pip install -r requirements.txt命令安装requirements.txt文件制定的相关模块,但本文是采用常用命令逐个安装各个模块的“pip3 install 模块名==版本 -i 源” Z% f; N- ~% d, `! s O, Z; z
- numpy==1.16.4, U( ^& H5 [8 E7 [+ w
- argparse' P/ Y, @, E7 M' G1 e8 p
- os
. [4 R( j: J0 d" S: c - logging4 f& X7 |6 _% |
- warnings
) Y) d8 V. y' G% W - datetime
; r) k$ P$ k9 r3 A - pandas==0.25.1
- Q; j, R+ f: C - scipy==1.3.1! Q6 G; H1 K8 X# R% i
- matplotlib==3.1.1- ~. G8 { s: j# k
- mpl_toolkits% b2 ?1 C! _+ _" v2 P
- sklearn-learn==0.21.3
}' l6 o! n3 q+ x; [5 q- I+ ^ - keras==2.2.45 V5 l- D; o! @) m
- tensorflow==1.14.01 n, g. l) Y O8 i- ~* j
- tqdm==4.36.1! y7 U% }+ V4 y+ [' g, B
- keras-tqdm==2.0.1
复制代码 j, C! }; v9 C% b# n
完成安装后,进入datasets目录,打开ReplaceWithWISDMDataset.txt文件,根据其提供的网址去下载8 e! X* B. D" P! p, S/ P
9 h& C i7 c9 f
. I* Y7 y& f% R$ Z* o' _0 [+ G, t7 M
+ _2 m- ^3 g0 R' g, b8 c/ w
下载WISDM实验室的数据集支持。
* ]0 c ]; j: v0 D) F: [% ]& I6 E, D& B, d; }9 w& |# o# D- u( u' l
- [" ]- t. o1 \# t7 ]3 w4 x, q# B
下载文件如下,将这些文件拷贝到datasets目录下覆盖。1 d d. g# ~" C- {$ K
5 Z' k) R3 y( M$ M3 ?' D4 l
9 c# ? e! d$ y# x: n0 t
& z( \( G. h- U2 z1 v. D. F
打开RunMe.py文件,可以看到关于各个运行参数的设置:
7 |/ a4 |" M* H3 z: K; V9 \6 y, ~( F2 b+ S$ A0 W
& \% @& T" R( p% g. Y5 N! v+ p/ A- m2 ~! J9 _, R8 e
运行python3 .\RunMe.py -h命令,查看运行参数含义,其中:--dataset使用的是前面下载的WISDM实验室的数据集来训练模型,而--dataDir是指定采用自行采集的数据集训练模型:
% W& X) O# N) i6 _- PS D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR> python3 .\RunMe.py -h6 I* D0 o4 P* m/ j$ w& |+ X
- Using TensorFlow backend.
! P+ {) O3 S. I f+ q - usage: RunMe.py [-h] [--model MODEL] [--dataset DATASET] [--dataDir DATADIR]* E8 A% X5 v# T, Z G6 ~( T. k
- [--seqLength SEQLENGTH] [--stepSize STEPSIZE] [-m MERGE]
$ y& x1 X9 U. _' a" m - [--preprocessing PREPROCESSING] [--trainSplit TRAINSPLIT]
4 M0 T' Y6 K) J; U" R* ` - [--validSplit VALIDSPLIT] [--epochs N] [--lr LR]: j0 B: C( x- p$ s q
- [--decay DECAY] [--batchSize N] [--verbose N] b% i- U; k2 _
- [--nrSamplesPostValid NRSAMPLESPOSTVALID]# L% c9 \- N$ c p0 k
-
2 r! ~1 u! B9 C' @6 u/ u& ] \ - Human Activity Recognition (HAR) in Keras with Tensorflow as backend on WISDM
# [: w1 G4 U: m A5 X' |8 J. H6 q: F - and WISDM + self logged datasets3 t/ O- V _7 S$ M- ]
-
* P2 w p8 [/ W( G: f7 {5 ] - optional arguments:* r7 r/ s& K0 p, [/ |, @5 `" C
- -h, --help show this help message and exit( i9 H: w4 |# A2 c, K$ }: L
- --model MODEL choose one of the two availavle choices, IGN or GMP, (( E3 H% W+ Z/ u1 x
- default = IGN )& h) \8 ]$ L& F& J; \6 ` ^
- --dataset DATASET choose a dataset to use out of two choices, WISDM or
3 ]4 _9 B- I$ P' W: |9 r5 z - AST, ( default = WISDM )
* Q, O# J1 R$ ^- L5 U - --dataDir DATADIR path to new data collected using STM32 IoT board
2 v6 u/ c% j) P+ k z- e, z% B - recorded at 26Hz as sampling rate, (default = )
- p6 C3 s+ M; u2 e( Z - --seqLength SEQLENGTH
& j: [. _* ^- R6 g - input sequence lenght (default:24)% l9 l8 f" j0 R% q
- --stepSize STEPSIZE step size while creating segments (default:24, equal/ h2 k& @# n: F0 a; z( ^, W
- to seqLen), Q$ d4 r2 h9 ~5 L( [/ @" m) i
- -m MERGE, --merge MERGE; b; e' ^/ Z. `7 f7 x
- if to merge activities (default: True)
0 l7 t6 P2 _3 v - --preprocessing PREPROCESSING
/ s& w3 |. J* s1 F9 K - gravity rotation filter application (default = True)- \- u8 J8 ?( ?, i+ i( O
- --trainSplit TRAINSPLIT
/ O: c' E% X1 I- g; _$ d/ m* S - train and test split (default = 0.6 (60 precent for5 Y6 O& i: E( U. o
- train and 40 precent for test))
% u! \' G) j! _9 [ - --validSplit VALIDSPLIT
1 F; T b/ F& |$ g5 D. V - train and validation data split (default = 0.7 (708 Y" ]+ i( A6 ~1 X% ^1 P1 p
- percent for train and 30 precent for validation))
9 v, U. g' t5 z - --epochs N number of total epochs to run (default: 20)
; w/ m& S! t) c - --lr LR initial learning rate
# P' j" p. Q Q6 I" e4 Y$ D' `" o - --decay DECAY decay in learning rate, (default = 1e-6)% R! o1 V8 y5 [( i
- --batchSize N mini-batch size (default: 64)
- Y( F. U& ]5 T- b - --verbose N verbosity of training and test functions in keras, 0,
/ G- C4 G$ r2 p1 W: I - 1, or 2. Verbosity mode. 0 = silent, 1 = progress bar," z2 R6 g& A8 u8 p$ j
- 2 = one line per epoch (default: 1)
. C: x* _( a0 A8 V0 ~. N# a - --nrSamplesPostValid NRSAMPLESPOSTVALID+ H8 l, T. t7 E% Q' N$ O2 o
- Number of samples to save from every class for post( t5 Y* {1 P* y. D+ m# c& ~' {& W w% h' o
- training and CubeAI conversion validation. (default =- Q2 p; k* s; T' j3 f& _
- 2)+ U, a& Z, {# F4 F
- PS D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR>
复制代码
5 T9 G3 o b+ g) z; t. x( T& Y7 w. O 在RunMe.py文件后面加入下面语句:
& R2 X7 D* s* X/ H& e. v- s- #保存Cube.AI支持的数据集格式,用于后续验证测试使用7 b* O6 u6 j7 ^$ x* F
- testx_f=resultDirName+"testx.npy"
2 A2 N. d' ?% M - testy_f=resultDirName+"testy.npy"; h2 ~2 z9 i1 v7 p
- 4 n6 N7 t0 q. c8 P9 U
- np.save(testx_f,TestX)
' w# T! w3 p3 u2 z( l/ ? - np.save(testy_f,TestY)
复制代码 - J* T1 f8 G# Q/ V# s
打开命令工具,输入命令python3 .\RunMe.py --dataDir=Log_data ,可以根据实际需要进行参数设置,本文先采用默认参数训练模型,输出日志如下,这显然是一个分类问题,分类为Jogging 、Stationary 、Stairs 、Walking,有卷积层、池化层、2全连接层、压平层、Dropout层等。& {! Z7 k8 d R( J; w" `0 h# [
- PS D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR> python3 .\RunMe.py --dataDir=Log_data7 t4 p2 F* A" t# ^6 J
- Using TensorFlow backend.' z. W8 {! z$ s: M/ m) {3 c2 {
- Running HAR on WISDM dataset, with following variables. R1 E; q7 D! C% P' z6 ~
- merge = True* v; @ J) j; O
- modelName = IGN,
/ |, F& b! b; t8 W3 x8 P' e' W - segmentLength = 24
* g, ]& O. h& r; |7 Y - stepSize = 24
; z6 b: `. R7 b0 X) ` - preprocessing = True
# Z, a) [) m" K( N. v - trainTestSplit = 0.6
% a9 w7 l, K& R9 E' } {# ? - trainValidationSplit = 0.7! O s1 j$ R! n/ _. U3 E- g
- nEpochs = 20" [" E- A) G( L, X, b
- learningRate = 0.00052 E% r3 {, C0 @2 z* l: S9 I
- decay =1e-06
5 N& v: T6 u: k2 C0 ]/ ?: {9 [ - batchSize = 64" b2 J0 Q3 m, j" {
- verbosity = 1
. R: F( g( F, {$ v9 P - dataDir = Log_data0 i+ O2 a: L% R
- nrSamplesPostValid = 2
% C/ q' E! J' M! X - Segmenting Train data
# u2 z9 K4 Y3 R7 h6 n ~* G6 e - Segments built : 100%|███████████████████████████████████████████████████| 27456/27456 [00:28<00:00, 953.24 segments/s]
8 @. q, t% P0 F$ I* }3 k - Segmenting Test data
; M$ r5 R) ~9 c# k2 K8 I - Segments built : 100%|██████████████████████████████████████████████████| 18304/18304 [00:14<00:00, 1282.96 segments/s]$ R3 ^" i- w( a% B, x- f/ A! R
- Segmentation finished!
) j; r5 b1 ` W& C. C - preparing data file from all the files in directory Log_data
/ V( f% H) f. g - parsing data from IoT01-MemsAnn_11_Jan_23_16h_57m_17s.csv
3 f1 P# w- r" M - parsing data from IoT01-MemsAnn_11_Jan_23_16h_57m_53s.csv
! e! F0 S1 E' {% D$ D- A - Segmenting the AI logged Train data
1 F/ T. V9 [2 x& K: K- w* k: |. @1 n - Segments built : 100%|████████████████████████████████████████████████████████| 25/25 [00:00<00:00, 3133.35 segments/s]& h1 q& u3 Q$ f E, _
- Segmenting the AI logged Test data
, c, _# n9 x* l- X1 c! V - Segments built : 100%|████████████████████████████████████████████████████████| 17/17 [00:00<00:00, 2852.35 segments/s]1 N7 |8 |) I5 S7 s5 h. _! W
- Segmentation finished!
8 B! L' z0 i( U" R. v: ] - _________________________________________________________________8 C' h1 y6 C9 ]0 p
- Layer (type) Output Shape Param #
0 r, @0 @5 P+ _' m7 v - =================================================================
# w5 }1 k# j' t) L5 N( M2 K* G6 { - conv2d_1 (Conv2D) (None, 9, 3, 24) 408
1 y1 y$ w/ X: [ - _________________________________________________________________% |7 w$ [; |1 A" Y9 f% c Z
- max_pooling2d_1 (MaxPooling2 (None, 3, 3, 24) 05 Z& c1 i e& ] U; \
- _________________________________________________________________
) ], C% \: G Z6 L - flatten_1 (Flatten) (None, 216) 0
2 G5 g, S( m& ^1 T) S! h' ]8 y - _________________________________________________________________7 B" _5 `2 g4 l
- dense_1 (Dense) (None, 12) 2604) S; Y* c' n A- J4 G0 E& m
- _________________________________________________________________
- x% b: B5 a' R - dropout_1 (Dropout) (None, 12) 08 v" u" P0 I$ a7 _
- _________________________________________________________________
8 h3 s8 S U. q/ g! T$ S( I# X - dense_2 (Dense) (None, 4) 52
# _0 I9 O' P l/ g: z; n - =================================================================
7 N: s1 E, t2 l' N5 a7 w5 K+ @ - Total params: 3,064& _; s- @& B. t: n' |# s) o5 O) k
- Trainable params: 3,064
$ O7 l: X1 c- u% K% C. d - Non-trainable params: 0
) B- K& X+ L. d: ]6 U& M: q- W - _________________________________________________________________
& J0 p/ V' I* o - Train on 19263 samples, validate on 8216 samples
d0 h4 f6 a; ^) J - Epoch 1/20
; j0 _) X' Z* Q: @: J3 G- g - 2023-01-24 14:41:03.484083: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
5 w9 l4 z/ B& R" @9 @& \7 h) P - 19263/19263 [==============================] - 3s 167us/step - loss: 1.1442 - acc: 0.5430 - val_loss: 0.6674 - val_acc: 0.7372
3 }/ Q0 C: n8 {& A, n0 P - Epoch 2/20( a* z- O: d6 f8 F8 t
- 19263/19263 [==============================] - 1s 40us/step - loss: 0.7173 - acc: 0.7089 - val_loss: 0.5126 - val_acc: 0.79284 O! E, j( y7 m$ A0 B& j) N: x+ }
- Epoch 3/20
2 ] c& R+ Q" } J; q - 19263/19263 [==============================] - 1s 40us/step - loss: 0.5954 - acc: 0.7522 - val_loss: 0.4470 - val_acc: 0.8051
2 ?& C( S5 a: d+ N' ? - Epoch 4/20
! o8 G4 R$ d f/ M8 b7 ^ - 19263/19263 [==============================] - 1s 39us/step - loss: 0.5288 - acc: 0.7810 - val_loss: 0.4174 - val_acc: 0.83354 ?" N& I4 f, X- ?
- Epoch 5/20
9 C* v0 @; v5 O' y9 @ - 19263/19263 [==============================] - 1s 36us/step - loss: 0.4925 - acc: 0.7994 - val_loss: 0.3897 - val_acc: 0.8477' C% P& A1 d/ O( C! X6 S3 t+ C, O& T5 h
- Epoch 6/20
2 i: {" d4 ?% O - 19263/19263 [==============================] - 1s 35us/step - loss: 0.4647 - acc: 0.8173 - val_loss: 0.3607 - val_acc: 0.8647
! }3 j+ x o# w* }% c - Epoch 7/20
p) R% }6 ` d6 f0 k% I - 19263/19263 [==============================] - 1s 37us/step - loss: 0.4404 - acc: 0.8301 - val_loss: 0.3493 - val_acc: 0.8777% T; ` K ^, p t4 D7 o* Y
- Epoch 8/206 [. x7 I( c. t6 H/ L9 V& y
- 19263/19263 [==============================] - 1s 38us/step - loss: 0.4200 - acc: 0.8419 - val_loss: 0.3271 - val_acc: 0.88271 R% P" K9 K6 y
- Epoch 9/20
1 |: f, s" S! N - 19263/19263 [==============================] - 1s 38us/step - loss: 0.3992 - acc: 0.8537 - val_loss: 0.3163 - val_acc: 0.8890: f2 u% V1 ^$ e8 y( Z2 j, e
- Epoch 10/20
- T, A9 O+ R( j: ~6 P - 19263/19263 [==============================] - 1s 40us/step - loss: 0.3878 - acc: 0.8576 - val_loss: 0.3039 - val_acc: 0.89918 ^' I# X9 D8 H. M2 Z8 Q
- Epoch 11/20
7 \$ A3 f" }0 @ - 19263/19263 [==============================] - 1s 40us/step - loss: 0.3799 - acc: 0.8667 - val_loss: 0.2983 - val_acc: 0.8985) Y- m W |, ^* I
- Epoch 12/206 `" V, i/ ?4 V9 ~5 V2 r5 r% z C
- 19263/19263 [==============================] - 1s 40us/step - loss: 0.3662 - acc: 0.8736 - val_loss: 0.2922 - val_acc: 0.9007% T: [1 i9 N" i# D3 n6 N1 d! b/ x/ T
- Epoch 13/20
% o, f, x. j7 Y& `" _+ y: G% q3 t - 19263/19263 [==============================] - 1s 36us/step - loss: 0.3613 - acc: 0.8760 - val_loss: 0.2837 - val_acc: 0.9051# ?# x# F: B2 o7 J' R% @; N
- Epoch 14/20
: W( i! w# N& K" a7 N7 s9 g ^ - 19263/19263 [==============================] - 1s 40us/step - loss: 0.3574 - acc: 0.8775 - val_loss: 0.2910 - val_acc: 0.8985- q, i# Y, W) V( ~
- Epoch 15/20; w3 f& P& H% e p( b
- 19263/19263 [==============================] - 1s 39us/step - loss: 0.3513 - acc: 0.8796 - val_loss: 0.2814 - val_acc: 0.9080
- m- p# I- J2 q; x - Epoch 16/20
: |: c& C1 q$ e4 O - 19263/19263 [==============================] - 1s 38us/step - loss: 0.3482 - acc: 0.8816 - val_loss: 0.2737 - val_acc: 0.9116
7 E! m* Q% X' q ^6 v - Epoch 17/20! |# m7 _ v* s6 \4 ?
- 19263/19263 [==============================] - 1s 35us/step - loss: 0.3362 - acc: 0.8875 - val_loss: 0.2742 - val_acc: 0.91144 {" ?% l) F+ ^; y
- Epoch 18/20
1 b9 h1 L: D6 X - 19263/19263 [==============================] - 1s 38us/step - loss: 0.3325 - acc: 0.8892 - val_loss: 0.2661 - val_acc: 0.9137
g) |0 @& W1 Z" o& X$ j6 R - Epoch 19/20
5 s9 X) O; l) r Q - 19263/19263 [==============================] - 1s 40us/step - loss: 0.3257 - acc: 0.8927 - val_loss: 0.2621 - val_acc: 0.9161% ^/ k/ ~3 q# T
- Epoch 20/20
& _# j8 C2 E; M2 D. Y# v0 h - 19263/19263 [==============================] - 1s 37us/step - loss: 0.3249 - acc: 0.8918 - val_loss: 0.2613 - val_acc: 0.9188- c9 n+ M* D2 F3 k' S
- 12806/12806 [==============================] - 0s 25us/step0 e+ Y# x8 D, r! e4 M
- Accuracy for each class is given below.
8 H* O1 d9 V; ?& \2 s - Jogging : 97.28 %
: j+ S% x' w2 j8 [" v( c$ Q3 N - Stationary : 98.77 %3 w* z ^5 E& v% C- [) s
- Stairs : 66.33 %2 {0 j( X0 ]+ z% \! W8 C
- Walking : 87.49 %5 y( \! E6 ?8 J8 Z. L0 z3 M; [) I1 G
- PS D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR>
复制代码 % @8 Y! _& q; H6 ?- F. ^
训练模型及相关输出信息在results目录下,每次训练输出依据时间生成一个独立目录,由于是keras训练模型,因此输出训练模型是一个名为*.h5格式文件,例如har_IGN.h5: p7 q3 Y* h+ k6 \0 F
- @& ]% X/ ~8 m8 ?( e
& u* ~' o6 l! D, z
/ s: D4 F6 D6 n
$ M$ j1 [) h# H4 S% u
4 W* ]7 l; X% A3 {8 a4 `
; ]# p8 N+ h8 P+ w& X- R
六、 cube.AI将训练模型转换为c语言模型+ R6 H: \# \( n* e4 \
6.1 创建cube.AI支持的STM32工程
3 r# f; y9 `& ~. l: F# g/ t 在CubeIDE中新建一个STM32项目,在cubeMX中选择以开发板形式创建
, H' b8 p+ G7 I+ o5 [, R4 u: t' I- C6 D7 t
5 g+ |& e- |+ D* S V
8 V0 _7 D& [0 N" o) j5 W ]7 @ 创建一个B-L475E-IOT01A_cube.ai工程名的STM32工程,如下图。# y) g2 p) c5 L4 v
' O4 J. u% e# {# X* M
$ s% w! t' N5 s' d: z% z
7 K$ p. C# H5 ?4 }+ B& W( G' M! i 完成创建后,双击.ioc文件打开cube.MX配置界面。
( l" S% B) o+ t+ s- P
: c4 @& R. W S' Q2 d/ W
5 g. W k0 K# i' c
# u/ U% }2 B- e; I$ D
2 ~3 a4 ?2 J& d# g3 J0 O% M# x 6.2 cube.AI神经网络配置: }2 H) `0 }8 b, ?( P0 r! p8 \5 _
选择X-CUBE-AI包支持,回到主页面后,会多出software Packs栏及多出STMicroelectronics .X-CUBE-AI选项,进入该页面,勾选下图标识的2、3 项,在5中选择采用哪个串口支持程序及调试。
* p7 \8 W) y( }; x- q5 y1 e2 B
9 e) m( f: q& a8 [% u' F$ s. x% i
/ w- {: J8 W7 ^7 T$ X, I* S7 d, }
. `2 l% @& {$ c7 J; H8 Z 知识点,在X-CUBE-AI配置选项页面,停靠时,会出现说明框,快捷键“CTRL+D”会进一步出现X-CUBE-AI相关文档,
; }# Y; _& ?# v" R2 w
0 H. i0 }% a9 `. U, d$ p: S# ^/ ]
! i n3 @" I( t! C0 O
+ I8 t2 \, v& D! N! R4 @" n 有详细的文档资料:
4 R: E5 y' l5 W3 M( I; K: t, p' E8 ]. H* P( s
. k) Z/ h Z- y2 m; A3 ~, U7 n
" n; k" F2 E2 a9 w$ I4 j 或者也可以从cube.AI安装目录直接进入,例如:D:\workForSoftware\STM32CubeMX\Repository\Packs\STMicroelectronics\X-CUBE-AI\7.3.0\Documentation& X: v% p7 h1 U* P% Y& }7 I+ |
# v. d5 m5 i8 Q: U7 n$ u 另外,需要注意,开启X-CUBE-AI支持后,其依赖CRC功能,会自动开启CRC。9 w b i7 E; A7 t& d; `( Z
, P+ u- [0 v! C) I4 P
, N% m( @0 }: b- m# L/ a
% g( S0 i: t; J- {" U% K7 v4 e 6.3 模型分析与PC端验证1 H2 H4 X. j6 G _4 W
添加(add network)神经网络如下,在3中可以修改神经网络模型名称,在4中选择支持框架及选择模型文件,例如“STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\har_IGN.h5”,在5、6中,可以选择随机数据进行模型研制,也可以选择生成的研制数据进行验证(前面训练模型时,在RunMe.py文件后面加入语句,输出testx.npy、testy.npy文件):
# m' |. z6 B, R, }4 @! f
3 ~$ Z& a. p" [! ~* d; L8 ^0 i8 H
' s+ u! ?: V/ m* F5 R- p/ ~
) @6 |. p- s) p# G; T7 u3 H
可以点击设置按钮进入,在该页面下可以对神经网络进行更多设置和更详细信息查看,主要是以模型优化为主,本文先保持默认。
2 G1 q$ f+ u' h- T, M; @# R, b; t. d" g: L, m- {9 F. m
( K: d' K7 @( t. \9 B$ z
/ E" |( Y/ T w5 C" s( k: H- @ 点击分析按钮(Analyze),输出该模型相关信息及部署模型需要的计算资源(ram、flash等):
( E r. h3 R# @1 D9 e
+ f8 P% o8 i S! ^, q-
0 @) o; p6 l( _. J" k$ c d1 { - Analyzing model , B0 S' v- v7 I/ Z5 ]7 V ~9 K; H
- D:/workForSoftware/STM32CubeMX/Repository/Packs/STMicroelectronics/X-CUBE-AI/7.3.0/Utilities/windows/stm32ai analyze --name har_ign -m D:/tools/arm_tool/STM32CubeIDE/STM32CubeFunctionPack_SENSING1_V4.0.3/Utilities/AI_Ressources/Training Scripts/HAR/results/2023_Jan_11_17_50_03/har_IGN.h5 --type keras --compression none --verbosity 1 --workspace C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace465785871649500151581099545474794 --output C:\Users\py_hp\.stm32cubemx\network_output --allocate-inputs --allocate-outputs % b1 G3 W2 z: ] i1 l/ r5 R
- Neural Network Tools for STM32AI v1.6.0 (STM.ai v7.3.0-RC5) \+ O& m: \7 h: Q; {, D
- ! ~* g( k3 d4 I. q
- Exec/report summary (analyze)
1 W8 S7 L& ?* R/ Q - ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
/ s) W$ @6 l. G$ l4 B3 |) { - model file : D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_11_17_50_03\har_IGN.h5 ; p# k6 f) Q# q6 x+ s
- type : keras , T$ A* @ A& ^; j4 s& h
- c_name : har_ign
1 f: g" W' _. H/ Y$ s- P - compression : none
( `& Z. Y: F0 N# m: q, l7 [ - options : allocate-inputs, allocate-outputs ; F$ t8 O2 o0 B, t# W' X
- optimization : balanced 7 Y9 ?* K- h9 x0 a
- target/series : generic
; k" @6 u0 o# W1 V5 V$ V: r; } - workspace dir : C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace465785871649500151581099545474794
6 ]( G/ {, k+ a3 U8 s - output dir : C:\Users\py_hp\.stm32cubemx\network_output ) }8 C4 a" v9 v, o
- model_fmt : float 8 ?) I D2 s; Z+ j! Q6 z7 _
- model_name : har_IGN : T8 a) n* a& M5 k! k; o
- model_hash : ff0080dbe395a3d8fd3f63243d2326d5 . N9 g; D( s. l W! f6 S5 v! _" f# M
- params # : 3,064 items (11.97 KiB) 6 Y0 H% k. Z( L& l6 j( P) g4 o
- ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
& j% v2 H7 h& M; f3 j# O( Y - input 1/1 : 'input_0' (domain:activations/**default**)
+ W& y4 H9 j5 ?9 j - : 72 items, 288 B, ai_float, float, (1,24,3,1)
# {4 V' Z8 `$ n. m - output 1/1 : 'dense_2' (domain:activations/**default**)
4 e, a" a# f! S4 Y) N. q: ~, V - : 4 items, 16 B, ai_float, float, (1,1,1,4)
; T5 m- m; O5 L% i% g - macc : 14,404
5 y# e( v' H9 |+ ?$ v - weights (ro) : 12,256 B (11.97 KiB) (1 segment)
$ ^2 S+ y4 U7 f& U7 F, s - activations (rw) : 2,016 B (1.97 KiB) (1 segment) * $ p0 u g, |5 e& [4 R: R( |
- ram (total) : 2,016 B (1.97 KiB) = 2,016 + 0 + 0 * h* ~& v6 h& c* J& y
- ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
* P+ m& \$ f# @# S6 T5 C* X8 H* \+ X+ x - (*) 'input'/'output' buffers can be used from the activations buffer
$ Q( F h$ U# i+ L! F4 ?0 Z1 v -
, N. N/ I# ~4 t; `; l7 R+ n - Model name - har_IGN ['input_0'] ['dense_2'] & `4 L) W/ ]1 a0 h! c5 B) e
- ------------------------------------------------------------------------------------------------------ 9 T# o- P3 B' g) i% r( @4 q0 ~, A% g% c
- id layer (original) oshape param/size macc connected to ; {& ]% F# U2 n4 @
- ------------------------------------------------------------------------------------------------------
1 V6 M# L- J9 k! m2 |3 L/ j& L2 u! n - 0 input_0 (None) [b:None,h:24,w:3,c:1]
q- _! x3 b2 ]6 A - conv2d_1_conv2d (Conv2D) [b:None,h:9,w:3,c:24] 408/1,632 10,392 input_0 : H$ K* ?& W& ~6 s
- conv2d_1 (Conv2D) [b:None,h:9,w:3,c:24] 648 conv2d_1_conv2d 9 m8 e3 d8 e2 ^$ K
- ------------------------------------------------------------------------------------------------------
& n' \1 E" J( r5 B# m" o/ ?& V: w - 1 max_pooling2d_1 (MaxPooling2D) [b:None,h:3,w:3,c:24] 648 conv2d_1 - q; N1 w$ u* b1 [* i. m" M
- ------------------------------------------------------------------------------------------------------
' I- Q) B# j7 Z" o2 F1 c% a - 2 flatten_1 (Flatten) [b:None,c:216] max_pooling2d_1
0 U7 N# q9 L1 O - ------------------------------------------------------------------------------------------------------
B$ t3 n& C) m0 l - 3 dense_1_dense (Dense) [b:None,c:12] 2,604/10,416 2,604 flatten_1
1 M3 w3 M, Q5 a - ------------------------------------------------------------------------------------------------------ : t' `6 H" e; A4 L* ^9 o
- 5 dense_2_dense (Dense) [b:None,c:4] 52/208 52 dense_1_dense # W& [8 u2 }/ K: ]( ?: s( S
- dense_2 (Dense) [b:None,c:4] 60 dense_2_dense . P: a+ I/ Y) w# \- h
- ------------------------------------------------------------------------------------------------------ 9 m. e6 _5 k' k
- model/c-model: macc=14,404/14,404 weights=12,256/12,256 activations=--/2,016 io=--/0
8 }8 [/ u' T! o- u -
. Y, _. s# ?% Y9 o/ o1 u' D( E$ f - Number of operations per c-layer # B* t! E6 m9 X! V9 T- q
- ----------------------------------------------------------------------------------- % P% @# H0 o, i# ]
- c_id m_id name (type) #op (type)
8 F9 @4 U# @; D; ~& y p - -----------------------------------------------------------------------------------
# W% E& f4 b+ z1 e4 K2 B - 0 1 conv2d_1_conv2d (optimized_conv2d) 11,688 (smul_f32_f32)
5 w! J1 Q9 b& j' v) h - 1 3 dense_1_dense (dense) 2,604 (smul_f32_f32)
9 n+ a' \0 \) J, ^/ X, v - 2 5 dense_2_dense (dense) 52 (smul_f32_f32)
' P; F6 Q, g7 L& q1 R, D; P4 w - 3 5 dense_2 (nl) 60 (op_f32_f32) 5 s# J# y: ?: F) `
- -----------------------------------------------------------------------------------
( `! y; O+ }- k8 q - total 14,404
3 T: P( l) f' L; l" } - 2 J+ `( V1 G+ @. G5 N- U8 I
- Number of operation types + ]" e+ Q8 c h/ O9 ~+ b% g
- ---------------------------------------------
" @$ q% ?8 ~$ {' F) k - smul_f32_f32 14,344 99.6%
. f* {9 k1 ^ V3 J) e - op_f32_f32 60 0.4% 5 g3 o% k# h5 p& a
-
8 W( ]' b, E" B2 _' J+ d* R4 y - Complexity report (model) ) R( C: o5 n: ]& `3 |- a+ ?
- ------------------------------------------------------------------------------------ 5 {& s2 Y+ X' p a
- m_id name c_macc c_rom c_id ; r' G2 l) ~) l+ W
- ------------------------------------------------------------------------------------
. \0 s! k: w2 I' d { - 1 max_pooling2d_1 |||||||||||||||| 81.1% ||| 13.3% [0]
$ ]1 `) e4 }& B) o4 ]: O2 v - 3 dense_1_dense |||| 18.1% |||||||||||||||| 85.0% [1] : |' U( G) T% P! h! n$ p
- 5 dense_2_dense | 0.8% | 1.7% [2, 3] 7 |* s2 I: {' H) {& _
- ------------------------------------------------------------------------------------ % ? ^. \7 @$ N K1 l
- macc=14,404 weights=12,256 act=2,016 ram_io=0 8 M: |! a; G5 J, q/ M4 I, I- l" g
- Creating txt report file C:\Users\py_hp\.stm32cubemx\network_output\har_ign_analyze_report.txt , k+ H. q% v( N) O3 q% g" u
- elapsed time (analyze): 7.692s 8 F0 S" a4 O$ G7 h7 m/ {7 j. k
- Getting Flash and Ram size used by the library
, n' Q2 o- ~/ J( I! J - Model file: har_IGN.h5 $ l: ?# B- j. Q8 M! [4 F
- Total Flash: 29880 B (29.18 KiB) / a! S6 t$ P. A6 s9 ?5 C
- Weights: 12256 B (11.97 KiB)
0 I- T/ I6 Y. M: H, [ - Library: 17624 B (17.21 KiB)
6 u/ I1 s# {) O7 R/ E& a1 J - Total Ram: 4000 B (3.91 KiB)
# A4 w2 D3 X/ e# X- B2 z9 B% A+ B - Activations: 2016 B (1.97 KiB)
2 R% x) }$ C5 p& o/ y - Library: 1984 B (1.94 KiB) & M. w" U. d1 }9 S1 P* N
- Input: 288 B (included in Activations)
5 s1 J/ Z" e. [ - Output: 16 B (included in Activations)
" ^+ B% W! t0 i* d - Done . x& w3 Y2 `" S9 x7 i G
- Analyze complete on AI model
复制代码
( X2 u7 f" q N7 G5 T# j" Z 点击PC桌面验证按钮(validation on desktop),对训练模型进行验证,主要是验证原始模型和转为c语言支持的模型时,验证前后计算资源、模型精度等差异情况,验证数据就是我们刚指定的testx.npy、testy.npy文件。& N0 w$ ~5 @& f1 i4 j1 g/ |* q! P( \
- ( |/ V* O# L( h$ o" v% \
- Starting AI validation on desktop with custom dataset : D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\testx.npy...
6 R% b0 d3 s- B @7 z3 C) _ - D:/workForSoftware/STM32CubeMX/Repository/Packs/STMicroelectronics/X-CUBE-AI/7.3.0/Utilities/windows/stm32ai validate --name har_ign -m D:/tools/arm_tool/STM32CubeIDE/STM32CubeFunctionPack_SENSING1_V4.0.3/Utilities/AI_Ressources/Training Scripts/HAR/results/2023_Jan_11_17_50_03/har_IGN.h5 --type keras --compression none --verbosity 1 --workspace C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace46601041973700012072836595678733048 --output C:\Users\py_hp\.stm32cubemx\network_output --allocate-inputs --allocate-outputs --valoutput D:/tools/arm_tool/STM32CubeIDE/STM32CubeFunctionPack_SENSING1_V4.0.3/Utilities/AI_Ressources/Training Scripts/HAR/results/2023_Jan_24_14_40_13/testy.npy --valinput D:/tools/arm_tool/STM32CubeIDE/STM32CubeFunctionPack_SENSING1_V4.0.3/Utilities/AI_Ressources/Training Scripts/HAR/results/2023_Jan_24_14_40_13/testx.npy
& `5 B2 X5 n' H - Neural Network Tools for STM32AI v1.6.0 (STM.ai v7.3.0-RC5)
! r* C$ U6 u! D) v3 Y. a" z2 ^/ `/ e - Copying the AI runtime files to the user workspace: C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace46601041973700012072836595678733048\inspector_har_ign\workspace 5 }. z$ ] P1 B
- . K6 D! F- u: X5 w; u5 z: r
- Exec/report summary (validate) & n/ R8 Y" G8 Z, }8 u3 a" G( w
- ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
" B& U6 s! M1 z$ g2 v9 ]0 [ - model file : D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_11_17_50_03\har_IGN.h5
i; a1 L& b4 T7 x9 y# a - type : keras
7 B. b. k* z2 _* ] - c_name : har_ign + {5 O6 c3 k; s
- compression : none
' _3 p7 ` S4 y7 q% Y - options : allocate-inputs, allocate-outputs
$ n9 a1 }0 W1 k0 U - optimization : balanced , P* U5 i& F3 S4 A$ A. D; M
- target/series : generic
- l3 x7 N, e! s# a1 O4 | - workspace dir : C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace46601041973700012072836595678733048
: e1 o; Z5 Y- V; l3 | - output dir : C:\Users\py_hp\.stm32cubemx\network_output ' |% C7 ~4 d" `2 S, z
- vinput files : D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\testx.npy
" W! P4 u+ G' s4 ~1 G/ u7 T - voutput files : D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\testy.npy ; T8 o; t* F3 r5 c' l' k( m
- model_fmt : float
# r t* ^$ ]) ]9 z6 h- g4 ]7 C - model_name : har_IGN
7 `! D# j F9 \ - model_hash : ff0080dbe395a3d8fd3f63243d2326d5 ) w" T T- ~; [" h
- params # : 3,064 items (11.97 KiB)
, [% z. r6 Z5 h+ R/ K - ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
; a$ _- c% B9 C' w6 F" C - input 1/1 : 'input_0' (domain:activations/**default**) 8 z3 \( d3 v# Z9 C$ p
- : 72 items, 288 B, ai_float, float, (1,24,3,1)
1 o( l6 ^) N" o7 o3 A - output 1/1 : 'dense_2' (domain:activations/**default**) b% d+ K, e+ q/ s* W: ~
- : 4 items, 16 B, ai_float, float, (1,1,1,4)
3 ]; K r: l( v7 n( S - macc : 14,404 ) _" T D' @+ R, f* A' d
- weights (ro) : 12,256 B (11.97 KiB) (1 segment) 8 G, k1 o9 z& r* g1 R2 q3 C
- activations (rw) : 2,016 B (1.97 KiB) (1 segment) *
+ z* h0 e/ w8 P9 C - ram (total) : 2,016 B (1.97 KiB) = 2,016 + 0 + 0
! |4 h# C+ _! i* x - ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
) j! S- `1 z1 w - (*) 'input'/'output' buffers can be used from the activations buffer - {. A2 Q: b: W; a) u3 s4 |% x
- Setting validation data... $ o" `9 k) e h$ c. e
- loading file: D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\testx.npy
7 l+ K2 K$ x* w" @. u - - samples are reshaped: (12806, 24, 3, 1) -> (12806, 24, 3, 1)
/ ^. w$ L$ n' {: q: |+ t ?8 I6 `: ]# g - loading file: D:\tools\arm_tool\STM32CubeIDE\STM32CubeFunctionPack_SENSING1_V4.0.3\Utilities\AI_Ressources\Training Scripts\HAR\results\2023_Jan_24_14_40_13\testy.npy
9 I# }) J) C5 ]- f, K8 n1 V" F+ v - - samples are reshaped: (12806, 4) -> (12806, 1, 1, 4) 4 E# ?- e$ G( U& s* W3 S
- I[1]: (12806, 24, 3, 1)/float32, min/max=[-26.319, 32.844], mean/std=[0.075, 5.034], input_0
, T( @) @7 i$ O/ i. \/ m - O[1]: (12806, 1, 1, 4)/float32, min/max=[0.000, 1.000], mean/std=[0.250, 0.433], dense_2 8 j* u5 |4 n3 j' Z" Q% S
- Running the STM AI c-model (AI RUNNER)...(name=har_ign, mode=x86)
+ i+ \) h* x) }# o& A - X86 shared lib (C:\Users\py_hp\AppData\Local\Temp\mxAI_workspace46601041973700012072836595678733048\inspector_har_ign\workspace\lib\libai_har_ign.dll) ['har_ign'] 4 _5 P y" ^9 m2 e. m
- Summary "har_ign" - ['har_ign']
7 i$ b6 |( t0 t; \: y - -------------------------------------------------------------------------------- ; }/ S+ p- @/ r
- inputs/outputs : 1/1 4 t# V) |1 `% o: ?$ F
- input_1 : (1,24,3,1), float32, 288 bytes, in activations buffer ) r+ V7 O$ j7 ~9 R; F7 c
- output_1 : (1,1,1,4), float32, 16 bytes, in activations buffer
4 K8 ~8 e: q! N7 _* p3 |! ]* F: C" ? - n_nodes : 4
; Y k6 |, t# X - compile_datetime : Jan 25 2023 22:55:51 (Wed Jan 25 22:55:47 2023)
8 C3 ]; C+ \# `: G - activations : 2016 / x7 m0 p( D$ ~. s3 K8 P
- weights : 12256 6 P7 T8 R" [9 w, q9 E6 Z; R/ y2 E
- macc : 14404 6 Z. \4 o* Q9 A2 z) d( b) ?
- --------------------------------------------------------------------------------
/ q) a, U! c% b8 i, \' y' C - runtime : STM.AI 7.3.0 (Tools 7.3.0) X/ S& X# _; {1 [( C
- capabilities : ['IO_ONLY', 'PER_LAYER', 'PER_LAYER_WITH_DATA']
5 r& c8 J N3 n! a+ ]( c R - device : AMD64 Intel64 Family 6 Model 158 Stepping 9, GenuineIntel (Windows)
" l1 F9 v) l. I/ \$ A$ P ] - -------------------------------------------------------------------------------- $ p/ e- p$ F7 v- E1 ]9 }
- STM.IO: 0%| | 0/12806 [00:00<?, ?it/s]
2 Z. Z" H* u5 o. B. S8 l0 R3 `: W - STM.IO: 11%|█ | 1424/12806 [00:00<00:00, 14136.31it/s] W, u% d4 n/ n. k8 c' ^0 F: P0 c
- STM.IO: 14%|█▍ | 1849/12806 [00:00<00:04, 2293.62it/s]
. K; S/ W/ Y s4 V2 D* c8 a - STM.IO: 17%|█▋ | 2170/12806 [00:00<00:05, 1774.74it/s]
: i) F) }( k6 F# x, J& A - STM.IO: 19%|█▉ | 2429/12806 [00:01<00:06, 1520.26it/s]
7 a: P- ]" i8 N" g2 I - STM.IO: 21%|██ | 2645/12806 [00:01<00:07, 1348.30it/s] , P4 l' L# e3 \: V# U* [# {
- STM.IO: 22%|██▏ | 2828/12806 [00:01<00:07, 1291.67it/s] }/ a1 i0 m: y: f$ W
- STM.IO: 23%|██▎ | 2992/12806 [00:01<00:07, 1245.52it/s]
2 Q% c: u I0 G0 F- I- w0 ] - STM.IO: 25%|██▍ | 3141/12806 [00:01<00:08, 1194.67it/s]
! c8 N- r+ @: C9 B }6 u7 e - STM.IO: 26%|██▌ | 3278/12806 [00:01<00:08, 1107.82it/s] ( E/ B0 Y$ U2 }" t1 f' H
- STM.IO: 27%|██▋ | 3407/12806 [00:02<00:08, 1154.55it/s] # E) G- ~: i) S
- STM.IO: 28%|██▊ | 3548/12806 [00:02<00:07, 1218.70it/s] ; ]( z' }- g7 h) C Q, l
- STM.IO: 29%|██▊ | 3678/12806 [00:02<00:07, 1175.60it/s] ) G( Q; }4 g/ D' X
- STM.IO: 30%|██▉ | 3811/12806 [00:02<00:07, 1215.67it/s] 9 q$ }, ~, G- J. l
- STM.IO: 31%|███ | 3938/12806 [00:02<00:07, 1139.58it/s] ; ~, E7 Z) D4 Q& @
- STM.IO: 32%|███▏ | 4075/12806 [00:02<00:07, 1197.83it/s]
9 d6 R+ t: ?- ^ N) V - STM.IO: 33%|███▎ | 4199/12806 [00:02<00:07, 1207.70it/s]
8 P9 y1 \# K/ k - STM.IO: 34%|███▍ | 4323/12806 [00:02<00:07, 1078.59it/s]
* B+ r/ F4 Z O8 h# y3 o - STM.IO: 35%|███▍ | 4451/12806 [00:02<00:07, 1129.92it/s]
1 T: Y0 x2 I: ^# x2 K: I - STM.IO: 36%|███▌ | 4590/12806 [00:03<00:06, 1194.76it/s] ) g3 h6 O3 ~/ b. {5 k
- STM.IO: 37%|███▋ | 4718/12806 [00:03<00:06, 1216.59it/s] ' w- q Y* l9 ~3 O
- STM.IO: 38%|███▊ | 4843/12806 [00:03<00:06, 1195.77it/s]
6 B9 ?/ |5 X9 a0 u3 v - STM.IO: 39%|███▉ | 4965/12806 [00:03<00:06, 1159.48it/s]
- `& L0 F' a }' _# H - STM.IO: 40%|███▉ | 5083/12806 [00:03<00:06, 1116.81it/s] 2 Z0 u( E _2 v' v
- STM.IO: 41%|████ | 5197/12806 [00:03<00:06, 1095.57it/s] ) c8 \+ L2 b# y7 i; T y+ Z! j
- STM.IO: 41%|████▏ | 5308/12806 [00:03<00:06, 1078.25it/s] 3 r+ Z. ~3 A0 Z( Z+ @$ ^
- STM.IO: 42%|████▏ | 5433/12806 [00:03<00:06, 1122.47it/s]
R4 n( Q# `' @7 W4 K/ R$ ` - STM.IO: 43%|████▎ | 5547/12806 [00:03<00:06, 1056.59it/s]
# a8 @. |4 g) w7 l# w- i - STM.IO: 44%|████▍ | 5655/12806 [00:04<00:06, 1055.01it/s] ) \3 @% P) X$ n% r% `% F
- STM.IO: 45%|████▍ | 5762/12806 [00:04<00:06, 1035.74it/s]
! L% _) s$ ?- B9 K$ C+ l( Q! h - STM.IO: 46%|████▌ | 5867/12806 [00:04<00:06, 1022.60it/s] 6 _0 f5 O1 a6 x, o7 ~2 \& e
- STM.IO: 47%|████▋ | 5981/12806 [00:04<00:06, 1053.06it/s]
+ I) W- f3 y% \! C/ \% S5 |4 M - STM.IO: 48%|████▊ | 6098/12806 [00:04<00:06, 1083.31it/s] ( i$ J2 Y; x; O& X9 ^7 l; N
- STM.IO: 48%|████▊ | 6208/12806 [00:04<00:06, 1025.35it/s]
+ l; _* ^, J1 K- Y2 q - STM.IO: 49%|████▉ | 6312/12806 [00:04<00:06, 952.27it/s]
1 M2 _7 A0 z( S4 m - STM.IO: 50%|█████ | 6410/12806 [00:04<00:07, 910.42it/s] 9 I1 O4 v" [7 @# n% H
- STM.IO: 51%|█████ | 6509/12806 [00:04<00:06, 930.92it/s]
- L0 o5 p; ]3 q7 e" I; K/ j" K u - STM.IO: 52%|█████▏ | 6620/12806 [00:04<00:06, 976.37it/s]
! r( J0 r q/ L# [( M3 P5 m* ? - STM.IO: 52%|█████▏ | 6720/12806 [00:05<00:06, 926.81it/s] + s' t) l4 m# c0 F; k
- STM.IO: 53%|█████▎ | 6818/12806 [00:05<00:06, 940.17it/s]
; N! X G8 Y y' t. [ - STM.IO: 54%|█████▍ | 6914/12806 [00:05<00:06, 930.36it/s]
- E" O1 L5 p+ [0 v' { - STM.IO: 55%|█████▍ | 7008/12806 [00:05<00:06, 852.84it/s]
5 u9 f- j; s: t- q# `) S- |. u3 a - STM.IO: 55%|█████▌ | 7106/12806 [00:05<00:06, 885.63it/s]
* x& [% @! I+ O! Y7 O5 J! ?* [ o/ z - STM.IO: 56%|█████▌ | 7197/12806 [00:05<00:06, 805.83it/s] n5 s- l' I% H- s
- STM.IO: 57%|█████▋ | 7299/12806 [00:05<00:06, 858.49it/s] 9 g# @1 Q% H3 a. z5 \
- STM.IO: 58%|█████▊ | 7388/12806 [00:05<00:07, 744.49it/s]
1 c* N0 o0 B$ I/ X1 K; m - STM.IO: 58%|█████▊ | 7473/12806 [00:06<00:07, 755.34it/s]
7 F5 ]; C9 e! Z0 V - STM.IO: 59%|█████▉ | 7560/12806 [00:06<00:06, 785.88it/s]
; K+ l8 b. ~9 B& | - STM.IO: 60%|█████▉ | 7642/12806 [00:06<00:06, 782.78it/s] . {+ p# x+ Y Q1 |2 @; q( U! b* d- r% t
- STM.IO: 60%|██████ | 7723/12806 [00:06<00:06, 768.90it/s] ( u( v& z) L' Z/ |! p
- STM.IO: 61%|██████ | 7825/12806 [00:06<00:06, 828.66it/s] 6 n% x! A: X4 P- f
- STM.IO: 62%|██████▏ | 7937/12806 [00:06<00:05, 897.30it/s] * |$ g# u2 C/ A+ Q1 S! R/ @! |
- STM.IO: 63%|██████▎ | 8033/12806 [00:06<00:05, 913.23it/s] - a) z' `6 U+ M9 d/ ^0 L6 W$ w
- STM.IO: 63%|██████▎ | 8127/12806 [00:06<00:05, 913.79it/s]
2 q7 Z9 I* A+ K0 |2 i9 w - STM.IO: 64%|██████▍ | 8254/12806 [00:06<00:04, 994.44it/s] ; I- t, S5 R; N3 \, C" X
- STM.IO: 65%|██████▌ | 8358/12806 [00:06<00:04, 1005.50it/s]
- W' s% t6 y) a6 n# ?7 X - STM.IO: 66%|██████▌ | 8466/12806 [00:07<00:04, 1024.62it/s] - P {7 Q- u- a: g
- STM.IO: 67%|██████▋ | 8579/12806 [00:07<00:04, 1052.03it/s]
% ?- [' O% i! R) }9 v - STM.IO: 68%|██████▊ | 8712/12806 [00:07<00:03, 1111.93it/s] & V+ ?% B# Q7 N( n" h3 }
- STM.IO: 69%|██████▉ | 8826/12806 [00:07<00:03, 1044.19it/s]
/ ]3 |7 R X+ x0 H( d - STM.IO: 70%|██████▉ | 8933/12806 [00:07<00:03, 1005.29it/s]
* d- R3 R4 n& }8 e5 T/ `8 }" U - STM.IO: 71%|███████ | 9036/12806 [00:07<00:03, 1010.21it/s]
: b2 `- v4 x! p- I - STM.IO: 71%|███████▏ | 9150/12806 [00:07<00:03, 1043.83it/s] $ M0 Z+ a9 ^* @0 X
- STM.IO: 72%|███████▏ | 9277/12806 [00:07<00:03, 1100.57it/s] 9 |5 C: k d. u9 ~6 Q
- STM.IO: 73%|███████▎ | 9404/12806 [00:07<00:02, 1144.16it/s] 4 `$ ?: A. p) i# J4 o: }- x
- STM.IO: 74%|███████▍ | 9521/12806 [00:08<00:02, 1135.98it/s] 4 x+ ^8 l' X7 t2 c( v' t: @
- STM.IO: 75%|███████▌ | 9648/12806 [00:08<00:02, 1170.75it/s]
9 k6 @: x0 s6 x" g - STM.IO: 76%|███████▋ | 9780/12806 [00:08<00:02, 1209.41it/s] / ~/ D. n$ S& }! C3 s* }
- STM.IO: 77%|███████▋ | 9903/12806 [00:08<00:02, 1184.92it/s] * R8 P: `% R! h' G% h& u
- STM.IO: 78%|███████▊ | 10032/12806 [00:08<00:02, 1212.12it/s] 1 L% I% f. H) n$ l0 @' v
- STM.IO: 79%|███████▉ | 10155/12806 [00:08<00:02, 1214.79it/s]
" w4 T# |+ B% v; c - STM.IO: 80%|████████ | 10278/12806 [00:08<00:02, 1096.01it/s] 7 d! Z( e* {6 w
- STM.IO: 81%|████████ | 10391/12806 [00:08<00:02, 1100.40it/s] * W- D7 N7 I* b; }7 R/ K+ X$ w1 a+ ~
- STM.IO: 82%|████████▏ | 10506/12806 [00:08<00:02, 1112.34it/s]
; G* H0 g6 S9 e4 [ - STM.IO: 83%|████████▎ | 10619/12806 [00:09<00:02, 1035.66it/s]
; y1 Z, b, `. M7 H, B: Z. O3 r3 T - STM.IO: 84%|████████▎ | 10725/12806 [00:09<00:02, 914.43it/s]
) j$ V; J" Y# \; [ - STM.IO: 84%|████████▍ | 10821/12806 [00:09<00:02, 889.74it/s]
& q- M5 z7 o5 r9 r- ^9 [" V9 T - STM.IO: 85%|████████▌ | 10920/12806 [00:09<00:02, 915.76it/s]
# T8 X! x8 V! R* a - STM.IO: 86%|████████▌ | 11014/12806 [00:09<00:02, 819.91it/s] # g& W( v4 d8 F8 h ^; \8 w4 j
- STM.IO: 87%|████████▋ | 11100/12806 [00:09<00:02, 738.28it/s] e9 A: Y" V a _
- STM.IO: 87%|████████▋ | 11178/12806 [00:09<00:02, 740.24it/s]
7 R; u9 y2 p; U; j6 k - STM.IO: 88%|████████▊ | 11255/12806 [00:09<00:02, 657.58it/s] 4 P: A9 J$ j6 ?& h6 H* P
- STM.IO: 89%|████████▊ | 11364/12806 [00:10<00:02, 702.16it/s] 0 a2 u" A2 n! h
- STM.IO: 89%|████████▉ | 11455/12806 [00:10<00:01, 752.49it/s]
/ Z( ?) ~" ^" H, f$ m* d - STM.IO: 90%|█████████ | 11548/12806 [00:10<00:01, 794.66it/s] - q; \9 H& Z4 \# p( V
- STM.IO: 91%|█████████ | 11631/12806 [00:10<00:01, 796.56it/s] 1 J) V% G) m8 ^3 A) a/ S
- STM.IO: 92%|█████████▏| 11748/12806 [00:10<00:01, 879.46it/s]
# Z6 M6 s: I0 I6 G ~) F4 I - STM.IO: 93%|█████████▎| 11853/12806 [00:10<00:01, 922.73it/s] % c( O* k5 Z' q B. l; Y
- STM.IO: 93%|█████████▎| 11949/12806 [00:10<00:00, 895.23it/s] ( h; \7 e) y) J8 u
- STM.IO: 94%|█████████▍| 12049/12806 [00:10<00:00, 922.41it/s]
3 m0 v# v7 N7 R - STM.IO: 95%|█████████▍| 12163/12806 [00:10<00:00, 976.60it/s]
5 a# k0 p; b( x$ e4 C - STM.IO: 96%|█████████▌| 12280/12806 [00:10<00:00, 1025.50it/s] ' y& H7 ]; R" Y8 M
- STM.IO: 97%|█████████▋| 12412/12806 [00:11<00:00, 1096.80it/s]
. M4 F* p3 L- t% q8 h7 a3 u0 |% w - STM.IO: 98%|█████████▊| 12525/12806 [00:11<00:00, 1072.91it/s]
' W/ E1 @! j3 ^6 F, K* U5 Z l - STM.IO: 99%|█████████▉| 12663/12806 [00:11<00:00, 1147.57it/s] , W E( K. @7 Z3 u+ N5 Q
- STM.IO: 100%|█████████▉| 12781/12806 [00:11<00:00, 1118.51it/s] - s/ v3 U7 Z0 W
- Results for 12806 inference(s) - average per inference
7 F) `0 B# k2 |4 q - device : AMD64 Intel64 Family 6 Model 158 Stepping 9, GenuineIntel (Windows) $ F. [* C" \4 Q
- duration : 0.057ms
* F/ M0 ~3 b2 h/ J2 j - c_nodes : 4 , b$ ~. f( ^8 K1 [ E& T
- c_id m_id desc output ms % 1 c* e0 E C& I. s! Q8 a3 X( v0 B& d3 E
- ------------------------------------------------------------------------------- 5 d( ~( y) @8 o5 Q2 S5 I' X# f, ?9 S
- 0 1 Conv2dPool (0x109) (1,3,3,24)/float32/864B 0.049 86.5%
/ d! _- @( ]$ h/ U9 F9 R - 1 3 Dense (0x104) (1,1,1,12)/float32/48B 0.005 9.1%
/ i$ c0 @& Q+ m0 f$ {( c. L - 2 5 Dense (0x104) (1,1,1,4)/float32/16B 0.001 1.8% 5 ?$ S0 q. `1 ~/ C5 Y: o5 l* I8 |9 R
- 3 5 NL (0x107) (1,1,1,4)/float32/16B 0.001 2.5% * O' B& I: E: H! W
- ------------------------------------------------------------------------------- + i l- t1 O. R+ `/ v7 `# Q3 N% i
- 0.057 ms
' A) Y& t6 h7 z3 J - NOTE: duration and exec time per layer is just an indication. They are dependent of the HOST-machine work-load.
, w) N" U2 }- u% U" R - Running the Keras model...
+ {( i9 B) H1 J( ?0 G* c% { - Saving validation data... # q: O* A9 E+ _5 O% o
- output directory: C:\Users\py_hp\.stm32cubemx\network_output % c9 _. u: J' [% }9 g3 o4 i
- creating C:\Users\py_hp\.stm32cubemx\network_output\har_ign_val_io.npz / N% M) r" m' H6 ]$ _
- m_outputs_1: (12806, 1, 1, 4)/float32, min/max=[0.000, 1.000], mean/std=[0.250, 0.376], dense_2
0 @* l! ]- C; O. C' u( n& @ - c_outputs_1: (12806, 1, 1, 4)/float32, min/max=[0.000, 1.000], mean/std=[0.250, 0.376], dense_2 ! c& N( G' u4 x: O) V4 i" B
- Computing the metrics... , ~4 \1 j2 i$ Y8 L0 S7 _) [9 X
- Accuracy report #1 for the generated x86 C-model
+ ?$ {- U# {" n5 M - ---------------------------------------------------------------------------------------------------- 4 }# I n6 q/ K8 }% D
- notes: - computed against the provided ground truth values ?7 o! [: x. E, [% b6 u
- - 12806 samples (4 items per sample) 8 f# N. |% [( Y- x& f
- acc=86.72%, rmse=0.224433631, mae=0.096160948, l2r=0.496649474, nse=73.14% 6 d5 A" q- X; }4 u
- 4 classes (12806 samples)
7 c l6 ?( N6 }; I/ @) ]0 u- L% A - ---------------------------- / t& R8 e* a" u/ I6 C( r
- C0 3678 . 62 41 3 X6 H( z6 X9 q' d* u, u2 p
- C1 . 1124 14 . 1 E8 G! ]: f5 |# W! L
- C2 254 10 1806 662
4 S9 E" r6 W8 O/ e* d7 _. B - C3 66 . 592 4497 $ S7 [% ?- Z) D1 v% W& {1 E
- Accuracy report #1 for the reference model 9 I0 N: v2 b: r3 V0 E
- ---------------------------------------------------------------------------------------------------- . s j) E4 f# Z& @, O7 A5 T; L
- notes: - computed against the provided ground truth values
" D: {- M L9 b - - 12806 samples (4 items per sample)
6 A# T% K Y/ h5 d9 l# D: M - acc=86.72%, rmse=0.224433631, mae=0.096160948, l2r=0.496649474, nse=73.14%
" A$ G2 @8 |" T w6 g9 a* q - 4 classes (12806 samples)
! B9 W0 h" X% k' L0 V$ ` - ---------------------------- % w; ?9 q* O J/ W- @5 y
- C0 3678 . 62 41 / y* y8 {9 q* R- `$ i
- C1 . 1124 14 . 2 U/ u/ j7 k5 c! g& g
- C2 254 10 1806 662 " x0 I( g M1 G% Z) [
- C3 66 . 592 4497
& M$ x% ^% l1 S1 ^* @' L5 C: g - Cross accuracy report #1 (reference vs C-model)
* q/ I9 ^& D2 G% n7 U - ---------------------------------------------------------------------------------------------------- " Z) B1 M) n- M$ @! B( ^ n3 Z9 W
- notes: - the output of the reference model is used as ground truth/reference value 0 Q9 ]: Z) c* _+ |
- - 12806 samples (4 items per sample)
6 u5 N% j8 r8 I: R$ n( R - acc=100.00%, rmse=0.000000063, mae=0.000000024, l2r=0.000000139, nse=100.00% 0 C4 i9 ?8 _; x: _
- 4 classes (12806 samples)
0 b0 i7 O+ e! m3 a$ C+ K - ---------------------------- 3 W' L. z! ?7 r r/ ~
- C0 3998 . . . & h( a3 A2 v2 H5 d, f5 p
- C1 . 1134 . .
5 j9 F; [6 U. [% |$ `8 Y4 E9 r3 _ - C2 . . 2474 .
t0 F' C% f: g6 V - C3 . . . 5200
& b3 o5 [: m& w: K -
) }& S8 G8 A, N% I - Evaluation report (summary)
2 v$ |* K. c( C: g - ----------------------------------------------------------------------------------------------------------------------------------------------------------
' \/ U+ O" F7 f Y% a0 ` - Output acc rmse mae l2r mean std nse tensor ' f3 x) x) x2 t e( J5 g5 A
- ----------------------------------------------------------------------------------------------------------------------------------------------------------
" D$ b. k" \+ U+ p- s; m% {$ \ - x86 c-model #1 86.72% 0.224433631 0.096160948 0.496649474 -0.000000000 0.224435821 0.731362987 dense_2, ai_float, (1,1,1,4), m_id=[5] 5 ` B5 P+ @: c; P ?1 a. H! q
- original model #1 86.72% 0.224433631 0.096160948 0.496649474 -0.000000001 0.224435821 0.731362987 dense_2, ai_float, (1,1,1,4), m_id=[5]
, Y- Q/ _& S2 b6 k8 b) J( Q2 [. E - X-cross #1 100.00% 0.000000063 0.000000024 0.000000139 0.000000000 0.000000063 1.000000000 dense_2, ai_float, (1,1,1,4), m_id=[5] $ ~9 l/ L, _/ r
- ----------------------------------------------------------------------------------------------------------------------------------------------------------
( `0 t4 \# @7 n - - w! S4 V$ t2 v% W9 x! b4 c8 x. D
- rmse : Root Mean Squared Error
9 o6 V1 H5 Q8 `/ ` - mae : Mean Absolute Error * M2 ^ I: }8 C3 c) H! j) ]! Y% b
- l2r : L2 relative error
4 v: k# n- B, D6 A ]- a - nse : Nash-Sutcliffe efficiency criteria # ]: U+ d5 c) R/ X8 f$ X' i( W
- Creating txt report file C:\Users\py_hp\.stm32cubemx\network_output\har_ign_validate_report.txt
- q+ }; x8 ^. U/ Z! K - elapsed time (validate): 26.458s + I5 h" Y0 Y9 Y* _8 j, G8 G/ g
- Validation
复制代码 . j3 ?/ d: U% c
6.4 c语言神经网络模型生成及源码输出0 @% u8 l9 `, ~! h" G# Y8 ^4 F
将开发板重新选择ST-LINK连接(5-6跳线帽拔出,插入1-2跳线中)
4 ]3 z% T* \2 v2 V+ V0 N+ n/ D( J$ N$ N: D) T" d; A( n8 ?+ G
为了后续源码讲解方便,只生产c语言的神经网络模型源码,不输出应用示例程序(有个弊端就是在新建程序加载到开发板后,validation on target功能无法使用),如下图所示。
h/ D3 ]# ~4 e8 ^ K- ^ e
. w' c2 F6 M( ^' p7 I0 Y: d. o' p* p g6 {
8 V+ `5 X6 q2 |9 c0 W
; b1 d7 T6 y& U E1 ?9 L 配置输出工程' K4 F+ T& ` W2 q$ i( L% V. a
# {& I% D$ N' e [
( F+ g5 e$ h, E+ U3 i% l
" v0 a4 p, f; y! F6 G5 o
七、c语言神经网络模型使用! d2 {$ T& [) C; V4 w4 o* V2 k
7.1 C语言神经网络模型源文件' K0 u* e9 C8 Z2 I
在cubeMX配置神经网络模型时,指明了名称是har_ign,会生成如下文件har_ign.h/c、har_ign_data.h/c、har_ign_data_params.h/c、har_ign_config.h这些源码文件就是转换后的c语言神经网络模型,提供了一系列的API,这些API通过调用cube.AI软件包的内置功能,工程实现了神经网络计算功能:+ c7 A6 B/ |1 s$ @. B9 u
/ ?+ w J! Q3 f V% ~
: L! r1 |2 P3 R* T( f 其中har_ign_generate_report.txt文件是生成c语言神经网络模型的过程记录。
% T- ?: ]1 S, \' ]0 e" o6 D7 g0 V' h
$ n- X( U, S, x* X
; j) }8 X& L- t
' E0 a9 Y- a4 { 7.2 串口功能自定义实现, M8 C; ?9 \+ V6 P- w9 @
由于本文没有选择生成配套的应用程序代码,因此串口功能还需要自己实现,因此我移植了串口功能代码,在工程目录下,创建了ICore源目录,并创建print、usart子目录,分别在两个子目录加入print.h/c和usart.h/c源码。
$ U; I( C1 {; W Z+ ] [
* T3 A4 U& }0 L2 L5 Z
) ~7 D5 w1 g$ o0 D) w/ c/ F7 v+ |* c( `2 k
print.h如下:
8 g0 e& V+ I' p7 B: F* |* U- #ifndef INC_RETARGET_H_
$ P2 R2 O) t7 Q$ m, v' v - #define INC_RETARGET_H_& @8 x; e$ X" k5 o
-
# y& Q2 H$ y m/ d - #include "stm32l4xx_hal.h"6 ~2 B2 W+ \6 [! S' E/ P/ y
- #include "stdio.h"//用于printf函数串口重映射4 _4 k) p7 Z2 `/ b0 H. X: K
- #include <sys/stat.h>+ P/ n: z1 M! @& w0 i) E
- % M) f% }2 p) `1 U
- void ResetPrintInit(UART_HandleTypeDef *huart);
9 K e2 n$ K2 d0 \$ J8 Q8 u - . f+ N; v' c1 N) E0 A' Y; M
- int _isatty(int fd);
* l" V2 J; b) H3 o# {( r. A - int _write(int fd, char* ptr, int len);, J/ g# l- p1 }2 `9 s- E
- int _close(int fd);
: n l9 W( g3 B/ v" h3 F - int _lseek(int fd, int ptr, int dir);% G5 t; k/ t( h2 l# Z
- int _read(int fd, char* ptr, int len);( y0 D5 n8 I) b- q
- int _fstat(int fd, struct stat* st);
- b- F4 u+ m% Y) s, c# H -
8 A* m& |+ K/ b5 n3 v' P - #endif /* INC_RETARGET_H_ */
复制代码
& E. T: ]- _! N+ f1 m2 Z. w' B print.c如下:
4 O# E. E$ Q5 X* x- #include <_ansi.h>
. U% K0 Q, b1 A2 |& v& f - #include <_syslist.h>
+ D: ]& c0 \9 ^6 I9 a1 B; s - #include <errno.h>
1 f8 l* \, \! q) S4 @4 s5 i6 l - #include <sys/time.h>
5 b9 W/ P1 p& Y) \. w& ^ - #include <sys/times.h>' J' j8 A4 U5 c( ~
- #include <limits.h>0 v% h+ W9 F k$ u4 J7 {6 M4 Q
- #include <signal.h>/ y |3 P# ~. a* I1 ]
- #include <stdint.h>3 N9 \% w% E. \7 d
- #include <stdio.h>
( T: f9 [$ b4 Y -
5 U- f# h5 N9 d1 t0 y" N6 D% M - #include "print.h"9 K" F1 y6 |+ `% e, c
-
3 r0 n2 j+ Q0 B9 E; D& Q8 J. a) ] - #if !defined(OS_USE_SEMIHOSTING)
, }* q3 {/ f3 z9 [9 O+ V$ p - #define STDIN_FILENO 0' V$ i' O7 b( m/ f
- #define STDOUT_FILENO 1
% H1 r4 _" P, u6 W/ W# J$ l: G4 T - #define STDERR_FILENO 2( Q; r! U' H! w
-
7 }. M. Y+ j8 G( x! C" j - UART_HandleTypeDef *gHuart;
" b) T5 i, D5 B -
8 M! }- W- Q; |+ i, ` - void ResetPrintInit(UART_HandleTypeDef *huart) {
) f" y3 k- Q$ B0 j( ]1 v, q+ u; Q - gHuart = huart;
/ Q1 Y/ i) j! W2 E0 D& O, ?' y - /* Disable I/O buffering for STDOUT stream, so that/ R) S# f4 a8 \. w
- * chars are sent out as soon as they are printed. */+ K! g0 F2 ]) x2 J! k/ K
- setvbuf(stdout, NULL, _IONBF, 0);
1 q) c- k" p6 c: C) Q - }- F3 c, \9 d% e9 L& N
- int _isatty(int fd) {
; H6 y) @/ }% l8 S5 {) x- x, ? - if (fd >= STDIN_FILENO && fd <= STDERR_FILENO)1 h$ E4 ~5 P( f7 D1 k6 H+ ^4 r
- return 1;
: s7 F1 B% F# a6 V - errno = EBADF;+ M+ O" F$ M6 e. c( d, u7 x
- return 0;( ^9 D+ a n$ e% E/ f* s
- }/ K1 J5 @! { z
- int _write(int fd, char* ptr, int len) {
" m: j+ o8 r( h$ B7 ` - HAL_StatusTypeDef hstatus;
' V3 t' \# y% l - if (fd == STDOUT_FILENO || fd == STDERR_FILENO) {
6 c W2 S5 f" p) Y' ~& S' [ - hstatus = HAL_UART_Transmit(gHuart, (uint8_t *) ptr, len, HAL_MAX_DELAY);' F1 [5 H8 F S6 L# g' F: F
- if (hstatus == HAL_OK)
) t! m: s% m e, o; a& E - return len;
) p1 o0 C0 r+ \" C - else! o0 I3 z; l0 r9 v8 a$ |
- return EIO;0 o9 z2 r) q# s! g5 I B9 _
- }
0 l4 b/ o. N2 L - errno = EBADF;7 c/ D9 u" C' L: [( J% p
- return -1;
. W1 u1 T: t. k6 z8 }# p - }0 t7 {" U% N( O) Q5 x* Q
- int _close(int fd) {4 c. [, d+ S& Y7 d- t
- if (fd >= STDIN_FILENO && fd <= STDERR_FILENO)# h b+ u+ \! h: x* f
- return 0;
/ t0 J5 o% S1 p x4 y8 r - errno = EBADF;% X- W+ D& n7 s5 D, J" z
- return -1;2 C6 Z; {% t. D
- }" H9 F; g# f" J
- int _lseek(int fd, int ptr, int dir) {
1 m I% z2 u6 t0 ?+ k W - (void) fd;3 O$ q3 a- t4 N' u) a1 X0 T
- (void) ptr;
/ v: s; C* Z& n/ o - (void) dir;, u7 {2 z: E3 k4 ^' ?
- errno = EBADF;
' U8 J! m. D! x) A1 ~ a. K2 _ - return -1;8 r' H0 M3 x( U/ u: q# p# d
- }
" A" k; U3 J4 `" L+ @# U - int _read(int fd, char* ptr, int len) {1 C0 o7 ^$ I. W l0 A2 m
- HAL_StatusTypeDef hstatus;
3 w5 U! S% v) s! N, R9 i. X - if (fd == STDIN_FILENO) {2 c# L7 A X% i. s' @! e# b5 }* p
- hstatus = HAL_UART_Receive(gHuart, (uint8_t *) ptr, 1, HAL_MAX_DELAY);; S9 i0 P7 G8 r$ [5 {! D" @
- if (hstatus == HAL_OK)3 m. O4 t( Y8 _
- return 1;
/ Q. e5 q+ ~9 P" G+ \( L ? - else
9 ~7 H) t- e+ A. |6 l, } - return EIO;
& T, a5 i7 t7 `1 u8 k5 F8 D! ^ - }% h% V9 Q3 B' }' W, d6 X, I. V6 [) u
- errno = EBADF;
) G- K8 [6 c0 w6 P7 } i0 u - return -1;
' P @7 u9 Q# z2 T3 z - }
. Y7 |( q8 W; U2 f5 j. l - int _fstat(int fd, struct stat* st) {! c4 U# U6 d3 G* o7 Z1 C* B
- if (fd >= STDIN_FILENO && fd <= STDERR_FILENO) {
' w. v* d7 P; D3 W# z* n - st->st_mode = S_IFCHR;( j( z9 B' K, X+ Z, E; {
- return 0;8 c3 I# c8 y9 Z. Z" P9 \) \! [
- }* a1 g4 s& O [
- errno = EBADF;
z Q, l- I# |5 r# o( p, Y - return 0;
% f* R4 T0 Q/ |4 z - }
2 @% D, V) J' E -
. ]: \, N* B. N* `6 j. h - #endif //#if !defined(OS_USE_SEMIHOSTING)
复制代码 6 N. T' A6 a& W- } A
usart.h
4 d4 w7 f- |/ N5 G- #ifndef INC_USART_H_4 Z9 \* `* r+ Z9 ~$ G J4 X, t! Q
- #define INC_USART_H_! x- p5 \( y8 q7 |7 {
- % |# j; O) C. c7 R8 p" w5 K
- #include "stm32l4xx_hal.h" //HAL库文件声明5 a, I. i* F6 x. M
- #include <string.h>//用于字符串处理的库/ V+ }7 D/ B! _2 O% W" q- c. n
- #include "../print/print.h"//用于printf函数串口重映射
/ y0 I- W( v1 w/ _/ A -
- m" R" w! v: T5 G9 T; _ - extern UART_HandleTypeDef huart1;//声明LPUSART的HAL库结构体% `5 m- k5 S$ s& c; f% V# b
- 6 H4 e# J( ^4 S, k3 m2 q* @
- #define USART_REC_LEN 256//定义LPUSART最大接收字节数
- e6 \0 G( V# q% s0 @ - 1 a' G3 K: Y4 M1 Q, e
- extern uint8_t USART_RX_BUF[USART_REC_LEN];//接收缓冲,最大USART_REC_LEN个字节.末字节为换行符
& R; h; w% u/ C- l - extern uint16_t USART_RX_STA;//接收状态标记
# z" k6 p8 Y M. c' K% I* f, m - extern uint8_t USART_NewData;//当前串口中断接收的1个字节数据的缓存! A ^. h5 b" v# C) g A
-
5 Q, C3 ? y" i -
5 Z G" |# e6 S3 {( |- Q1 Z1 z - void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart);//串口中断回调函数声明
% r, V# p2 r0 l# I% g -
7 A' {' n4 B( d2 b+ o1 ~5 S - #endif /* INC_USART_H_ */
复制代码
* P' p- I) R, i/ z) |2 J6 s# C usart.c如下:
' y8 T0 W! A0 ~+ l9 ^5 L- #include "usart.h"1 r& z( q% [" b" R5 h, G0 d+ z' c8 F5 |2 s
- : |- H0 `# L6 x$ Y7 B3 ]% S1 n
- uint8_t USART_RX_BUF[USART_REC_LEN];//接收缓冲,最大USART_REC_LEN个字节.末字节为换行符
9 p' W: K Q1 x. n- S2 s$ u- b# Q - /*
! j9 p* E: @& e) O - * bit15:接收到回车(0x0d)时设置HLPUSART_RX_STA|=0x8000;
+ R; ?4 K2 H# d9 L/ s5 D - * bit14:接收溢出标志,数据超出缓存长度时,设置HLPUSART_RX_STA|=0x4000;
4 f7 A, B, y1 t7 ?1 a' t4 o - * bit13:预留! W& f8 z- W6 l1 C! `6 M z* e6 _5 k3 d
- * bit12:预留2 ~+ i# q1 T/ @. x
- * bit11~0:接收到的有效字节数目(0~4095)
! _: n& |+ m9 I9 s - */" C) ]1 U0 p ~& d3 t
- uint16_t USART_RX_STA=0;接收状态标记//bit15:接收完成标志,bit14:接收到回车(0x0d),bit13~0:接收到的有效字节数目
; L+ B [$ W2 k) h" F; e - uint8_t USART_NewData;//当前串口中断接收的1个字节数据的缓存3 m. B: o: r6 I+ o# B
-
V. C: E2 c# J - void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)//串口中断回调函数0 ~ r! _' Y7 n9 [9 Y0 L9 {
- {
1 C6 G. {8 w/ A' V8 } - if(huart ==&huart1)//判断中断来源(串口1:USB转串口)
6 a% w4 U1 p, } - {
# ~' K/ [" G3 h. a - if(USART_NewData==0x0d){//回车标记$ y/ Y: C) \ C1 t% h
- USART_RX_STA|=0x8000;//标记接到回车
J9 X' r& Z2 H; ?2 F; S: J - }else{
5 C2 y S+ `- M - if((USART_RX_STA&0X0FFF)<USART_REC_LEN){
6 G7 W ?4 s8 {8 {9 P0 q - USART_RX_BUF[USART_RX_STA&0X0FFF]=USART_NewData; //将收到的数据放入数组- V7 }/ f. D' G9 z) G" X3 ^
- USART_RX_STA++; //数据长度计数加16 I3 r+ T' q S, O. M1 n' P; |. o
- }else{
% R: ?/ R$ w, u! P/ Z+ B: } - USART_RX_STA|=0x4000;//数据超出缓存长度,标记溢出+ l0 x2 }* Q8 I2 T
- }
' P8 [7 ~0 g) P& G- K5 } - }
& n# z* A: v7 A - HAL_UART_Receive_IT(&huart1,(uint8_t *)&USART_NewData,1); //再开启接收中断. [2 t0 \3 n! Q. w( S- Q; S: t
- }0 q+ }" G. a! P' h, U
- }
复制代码
# S9 g2 B/ u S. i# j# @ 7.3 c语言神经网络模型API使用
" W8 b6 l" S% u+ d0 `( |& r 先不管底层机理,本文给下面代码,看看如何实现这些API调用的,在main.c文件中,通过aiInit函数,实现har_ign模型初始化,并打印模型相关信息。在主函数循环体中,通过串口输入信息,获得数据生成因子,调用acquire_and_process_data进行输入数据生成,然后调用aiRun,并传入生成数据及输出缓存,进行神经网络模型调用。然后调用post_process打印输出信息。2 l; o$ }3 s1 [
- /* USER CODE END Header */
3 W7 [( ]2 V0 A) r0 Z, R" x - /* Includes ------------------------------------------------------------------*/6 ~/ F7 U' \3 a) x' l8 v( a) z
- #include "main.h"5 U5 G) X0 R) h; F
- #include "crc.h"3 A; d8 m& i d$ }
- #include "dfsdm.h"# o0 ?& ?6 S6 B* C0 C
- #include "i2c.h"6 U% s9 p( M7 _/ z0 B. S
- #include "quadspi.h"; p7 r# o0 ?9 P2 f: i5 l7 U$ D8 x
- #include "spi.h", ?# y) Z+ x6 _9 l$ B* F0 y" e# |
- #include "usart.h"8 ^$ O; m) v# c/ r8 ?1 y0 w
- #include "usb_otg.h"# [% y; K# `0 n) D& o
- #include "gpio.h"9 [. p1 _' z( Z
-
7 W$ A1 d" I) M1 V7 t% _" \4 s - /* Private includes ----------------------------------------------------------*/$ O6 C( x! ^7 A e
- /* USER CODE BEGIN Includes */
4 v. b, v, [3 m6 Q; i - #include "../../ICore/print/print.h"
4 q; l: [, Z, R0 W - #include "../../ICore/usart/usart.h"7 i/ G8 v6 _% `! z9 t
- #include "../../X-CUBE-AI/app/har_ign.h"' F" t7 l+ ~( K
- #include "../../X-CUBE-AI/app/har_ign_data.h"2 Q+ A4 U6 W: K& q" {# ]+ D+ }
- /* USER CODE END Includes */
' ?9 `. j% U9 b7 o - / Z& J6 U. Y9 [* h9 l
- /* Private typedef -----------------------------------------------------------*/' D, E1 n( i' K' A% x
- /* USER CODE BEGIN PTD */1 I8 i) @1 J2 @1 H
- - c6 ?$ {0 F! \, p& ?, N7 K
- /* USER CODE END PTD */8 \" m7 z c/ G) f3 U# H# M
-
, m' V, ~' H0 B0 y( Y. Q+ x - /* Private define ------------------------------------------------------------*/% r5 P' Z1 K6 O9 i' W' V
- /* USER CODE BEGIN PD */
! p: O, I, y; u0 `' u - /* USER CODE END PD */+ P$ a1 z. E5 Q! j5 g8 ~- E
- + N3 @9 Z4 M# j/ h
- /* Private macro -------------------------------------------------------------*/0 d; X$ u9 B& }
- /* USER CODE BEGIN PM */) O$ k$ Z0 Y( z% k# x$ a: ?" q
- * F5 E+ |0 Q) K) U
- /* USER CODE END PM */
5 i( N% O, I' J' E2 ] -
- U$ q6 S: `+ n1 _, C4 H - /* Private variables ---------------------------------------------------------*/
; X& |; R: B( i. Y; f - 8 N6 j9 K( F8 F4 ^6 |0 K/ ^: d
- /* USER CODE BEGIN PV */! D7 Z* C8 f4 O# _. Y9 q
-
: c/ R" u) B& J( c& ?9 ~ - /* USER CODE END PV */
6 V# K) D W0 Y" T' Q, { - 4 \8 o* _: R; R0 V! t% Y3 I
- /* Private function prototypes -----------------------------------------------*/3 S C/ B" D" { G: P
- void SystemClock_Config(void);
4 \/ L# p% v# r/ ]# q# \ - /* USER CODE BEGIN PFP */: ~8 r8 F J5 q
-
, s/ f8 f& e; ?; [" X2 z - /* USER CODE END PFP */
1 Z& e1 |( ? \% r, R/ a' O5 v -
( Q" x+ P+ k. g/ v - /* Private user code ---------------------------------------------------------*// u; I6 s. R& F F3 R* y
- /* USER CODE BEGIN 0 */
8 Y- J2 Z% F& L O4 X P/ g - /* Global handle to reference the instantiated C-model */
) Y' f9 C$ [1 g/ F$ b G - static ai_handle network = AI_HANDLE_NULL;
, g& f' g6 B* w( w; I( C$ G - 2 I; ~$ y0 ]- J5 i3 l8 X+ X) C
- /* Global c-array to handle the activations buffer */# U: S7 K7 t4 _- f
- AI_ALIGNED(32)
a- h( }5 c+ C$ F' s - static ai_u8 activations[AI_HAR_IGN_DATA_ACTIVATIONS_SIZE];
) Y+ Y; Z/ {6 t' ^ -
5 v; C* E5 T7 v5 d; l - /* Array to store the data of the input tensor */( Q; C# V1 ^; F! }$ u
- AI_ALIGNED(32)
H* ?* G/ |9 c7 {# D: y - static ai_float in_data[AI_HAR_IGN_IN_1_SIZE];
O( N! X# a- y - /* or static ai_u8 in_data[AI_HAR_IGN_IN_1_SIZE_BYTES]; */
" ]" y6 a' ^; ~$ W9 b1 J/ P" N -
5 n$ y4 D6 e6 {3 }# |% w' w - /* c-array to store the data of the output tensor */3 S1 [: ^0 C3 L% e7 I* m
- AI_ALIGNED(32)
0 ^/ ]7 ~. U8 I; C - static ai_float out_data[AI_HAR_IGN_OUT_1_SIZE];( J; W9 |, x8 R5 a( S
- /* static ai_u8 out_data[AI_HAR_IGN_OUT_1_SIZE_BYTES]; */
* }5 c& p) O; k: r9 |7 E$ h - ' z2 s$ c- r1 V) ^
- /* Array of pointer to manage the model's input/output tensors */
- n; ~5 H7 _. m! i+ o5 P - static ai_buffer *ai_input;3 }' z% K0 A6 Z/ x9 m, Y- i
- static ai_buffer *ai_output;
1 J0 B, z8 _+ P! f6 j5 s - static ai_buffer_format fmt_input;2 E% X( D" D. y" _* o3 l2 T
- static ai_buffer_format fmt_output;: w5 z* `- O8 [* b
-
a+ p, W9 Z2 j6 Y - void buf_print(void)
+ J' U* R0 X* H& B' a# O/ n" o! n - {
7 A0 j4 Y& O* }( q - printf("in_data:");
6 P$ U% r5 |: \+ i - for (int i=0; i<AI_HAR_IGN_IN_1_SIZE; i++)% J' U) `0 M8 z$ T7 X; X
- {
* m' p7 S: ]7 b& J - printf("%f ",((ai_float*)in_data)[i]);
$ y8 I! L. x8 r& L - }
* k% G& T# t8 X- ~% m8 I5 R6 D - printf("\n");
" f% v/ m* H% A& _& P N - printf("out_data:");
# \! v n# \% T# U - for (int i=0; i<AI_HAR_IGN_OUT_1_SIZE; i++)
3 S! a9 G; y" F3 O* Z; X - {7 [* i; g( i" m( ]1 M7 q6 [8 S5 ~$ t$ E
- printf("%f ",((ai_float*)out_data)[i]);( `3 S5 Z, D# Y
- }! L) ^3 ?. r( S- c% r2 {& V
- printf("\n");/ W( W" H' c5 s
- }
' i* |7 i0 M0 }* K' N - / F. g/ Z) C& ^. }7 m! D* x
- void aiPrintBufInfo(const ai_buffer *buffer)4 w) Q9 O8 B# ? \* t
- { E' e" Z; M- t+ v8 Y
- printf("(%lu, %lu, %lu, %lu)", AI_BUFFER_SHAPE_ELEM(buffer, AI_SHAPE_BATCH),
; W; ^& C0 `& v& e4 m: p - AI_BUFFER_SHAPE_ELEM(buffer, AI_SHAPE_HEIGHT),/ a% Y* R2 V1 e# j9 t, T
- AI_BUFFER_SHAPE_ELEM(buffer, AI_SHAPE_WIDTH),& U o+ O6 o4 I0 o8 k% e
- AI_BUFFER_SHAPE_ELEM(buffer, AI_SHAPE_CHANNEL));3 ^; i* }; Z @
- printf(" buffer_size:%d ", (int)AI_BUFFER_SIZE(buffer));
* ^+ {- c/ h/ M! | - }" O' B. R" C: e% s8 W- h
- 0 p) L1 [) d: Q
- void aiPrintDataType(const ai_buffer_format fmt)
8 V2 w- G' x) o1 B2 _& H7 B/ ^ - {
4 y' A8 [. f) M* ?" Y+ r; b - if (AI_BUFFER_FMT_GET_TYPE(fmt) == AI_BUFFER_FMT_TYPE_FLOAT)
; j: J" H9 t+ }5 J# U - printf("float%d ", (int)AI_BUFFER_FMT_GET_BITS(fmt));
8 j! c+ [: S/ r; G* q1 p - else if (AI_BUFFER_FMT_GET_TYPE(fmt) == AI_BUFFER_FMT_TYPE_BOOL) {
7 S: n5 T- O; a( ~: ~$ ~$ w - printf("bool%d ", (int)AI_BUFFER_FMT_GET_BITS(fmt));
5 h- G# `- F8 e - } else { /* integer type */
# G# ]% L0 d% ]) `, x( }% K - printf("%s%d ", AI_BUFFER_FMT_GET_SIGN(fmt)?"i":"u",& S; y; @5 g/ s6 U+ d4 Q7 g$ }3 ?
- (int)AI_BUFFER_FMT_GET_BITS(fmt));
8 o4 L' h) O- j4 h. a - }( `( j. M9 w. h
- }
3 s1 z) F! H! t/ r* A3 Z) M% ?- W -
* F |% ^( e: r - void aiPrintDataInfo(const ai_buffer *buffer,const ai_buffer_format fmt) t4 {0 D! H! y% I/ D. M
- {7 U! s0 Q9 p; q( [& k0 s1 D: i3 o3 T/ y
- if (buffer->data)6 W+ Y' Z! o, ~) F2 K- q: K( i
- printf(" @0x%X/%d \n",1 v/ ]/ c" e m8 p! i2 }" i
- (int)buffer->data,
' f( Y" Y, A$ |; h5 b9 o( E5 A - (int)AI_BUFFER_BYTE_SIZE(AI_BUFFER_SIZE(buffer), fmt)
6 C# R2 O8 R* C) {( v - );. b2 h, S- n$ u2 z5 B( n7 N
- else* T6 T; A0 M; t+ w w" }
- printf(" (User Domain)/%d \n",. ]" I7 S& U8 o4 M2 D8 q1 y; K+ L
- (int)AI_BUFFER_BYTE_SIZE(AI_BUFFER_SIZE(buffer), fmt)% S6 x/ \) x7 K) @
- );% ]& {6 y6 B! T8 S
- }
6 o5 q1 N1 C, w6 n2 E1 q1 ^; ? -
, r% l- m# Q+ v7 j - void aiPrintNetworkInfo(const ai_network_report report)
: v6 w! p+ h! Y; H - {" I) Q8 y, X' p
- printf("Model name : %s\n", report.model_name);* I2 n9 `# ]9 X+ G( ?
- printf(" model signature : %s\n", report.model_signature);
& J' N! x; f# R4 i - printf(" model datetime : %s\r\n", report.model_datetime);
7 R+ F! @( z% x0 a g1 @ - printf(" compile datetime : %s\r\n", report.compile_datetime);* T5 T+ K Q! a) R
- printf(" runtime version : %d.%d.%d\r\n",
* }$ i, B; u5 Z! F+ } - report.runtime_version.major,
, @7 i3 _6 d( E6 P* x/ R - report.runtime_version.minor,
" q, _2 w9 B+ y. | - report.runtime_version.micro);
( x' v9 x" r& ]1 S3 O - if (report.tool_revision[0])
6 W" q8 O4 e q1 f5 j6 i - printf(" Tool revision : %s\r\n", (report.tool_revision[0])?report.tool_revision:"");
) n" J G8 W; L9 b; F# p2 @9 } - printf(" tools version : %d.%d.%d\r\n",
& \8 L# u: O" l - report.tool_version.major,
& L4 o, Q. m9 g" v3 P s5 O# v - report.tool_version.minor,
8 ~+ A2 `; P4 W5 R2 G) _ - report.tool_version.micro);8 e/ P! z% w2 H O0 L: ]) S
- printf(" complexity : %lu MACC\r\n", (unsigned long)report.n_macc);
8 t3 t. w; f0 D# g- p d - printf(" c-nodes : %d\r\n", (int)report.n_nodes);- g; X; Q. l' e/ I0 K
-
! }9 c: c& f4 R5 Y - printf(" map_activations : %d\r\n", report.map_activations.size);
x5 U* i. p$ _. R7 n8 _/ _ - for (int idx=0; idx<report.map_activations.size;idx++) {
0 |6 p8 r8 i5 E6 S& n0 P - const ai_buffer *buffer = &report.map_activations.buffer[idx];) u1 x, G1 K% u6 k. c, L( q
- printf(" [%d] ", idx);. X4 G+ E0 Y9 I! Y' \7 @" s2 G
- aiPrintBufInfo(buffer);. D, r$ \* ~0 v" f
- printf("\r\n");
* L4 n# h% ~$ n8 H8 U; Z - }
J/ F6 `# ~! p - * k( x9 Y* |& C t: x
- printf(" map_weights : %d\r\n", report.map_weights.size);
; r& e. Y! W' [1 I D- \2 d - for (int idx=0; idx<report.map_weights.size;idx++) {
) C2 o' N8 C* W3 v5 N; l: K9 W2 d - const ai_buffer *buffer = &report.map_weights.buffer[idx];: _/ r B; l) A
- printf(" [%d] ", idx);
/ I2 S+ C- D c$ m - aiPrintBufInfo(buffer);
8 z. w- b" s' j3 B: ?/ n. o - printf("\r\n");9 ]- d" ?* b1 C Q& w4 Q$ I
- }; G: `$ T* O& x4 E3 s
- }4 h, r2 W. f* W! r3 W6 f p
- 8 X- ~0 l) J- W1 d
- /*
" ^( O8 R9 s" D+ u: V+ p - * Bootstrap7 _3 y! F4 m3 X9 O1 A' C
- */( U; U3 H9 b; ]) H
- int aiInit(void) {3 g5 j* n5 r7 }" ]
- ai_error err;
6 e0 i* o J; s k - * ?% }- [( Q- B/ D d
- /* Create and initialize the c-model */1 H' w5 I7 M: l A# s. K0 \
- const ai_handle acts[] = { activations };
7 l8 R) t* u! M2 S/ d: c% F - err = ai_har_ign_create_and_init(&network, acts, NULL);
) v7 {+ a( ~; l - if (err.type != AI_ERROR_NONE) {
/ q3 t! [$ A& a7 V+ I5 p8 l! H - printf("ai_error_type:%d,ai_error_code:%d\r\n",err.type,err.code);8 H8 i: b" b7 s& ^
- };
! f8 H$ m5 O" h% A. W" e- p -
- Q0 a6 M' o7 G: \ - ai_network_report report;
8 c, p- {7 G, w# P - if (ai_har_ign_get_report(network, &report) != true) {
( k) k4 Z* D/ l; b - printf("ai get report error\n");+ J, `+ {' G2 U3 R
- return -1;: n9 @9 |( y8 {2 S) i, x, \9 l9 M
- }. l9 r$ H1 ^# ~4 j/ Q8 H
- ' b) {: I; Z) ]7 V9 @# I
- aiPrintNetworkInfo(report);
- \/ j8 U0 t( K( a' J" o - 6 t6 L6 E' c* Z+ |2 e
- /* Reteive pointers to the model's input/output tensors */" ]5 T, u/ `; l3 j {
- ai_input = ai_har_ign_inputs_get(network, NULL);& j8 C3 A+ n9 F+ F" J# m
- ai_output = ai_har_ign_outputs_get(network, NULL);
! o# }' n: ~8 Q) a - //! \# |3 f* g! }' b4 `# ^
- fmt_input = AI_BUFFER_FORMAT(ai_input);
2 P& I9 ^% z8 J3 p! |" I6 v9 p2 d' F - fmt_output = AI_BUFFER_FORMAT(ai_output);) ]: d, s" ?3 P# N9 V6 Q$ ^
-
5 Z* F" B0 D% u9 f0 I4 ~- k - printf(" n_inputs/n_outputs : %u/%u\r\n", report.n_inputs,7 Z* t3 [: z0 }' x) q: c
- report.n_outputs);
; W0 O2 G z/ {& e e - printf("input :");* t! G; M" | {7 _. q5 I
- aiPrintBufInfo(ai_input);
4 c! L$ A. j0 ?* Y D9 T6 C7 p2 F - aiPrintDataType(fmt_input);
0 w! T4 m& v) W& c$ f - aiPrintDataInfo(ai_input, fmt_input);
' u4 W- e0 ]: s5 j% P9 M @ - //# E _* V0 H: J5 U" f E
- printf("output :");
3 ?5 Q2 f5 H2 F9 G( m% X - aiPrintBufInfo(ai_output);
$ o1 x! l% u. J- P - aiPrintDataType(fmt_output);
B, S) y/ D& r) @ - aiPrintDataInfo(ai_output, fmt_output);# K& I4 x, d. J2 h, }- j, F% j( l
- return 0;
* V6 J+ E# d- C K$ p - }
# U u, q8 V* ?# b -
5 ~0 Z0 l1 i3 f/ N3 K# E+ _. D - int acquire_and_process_data(void *in_data,int factor)
& a- e3 M) j5 ^1 J+ P) U - {
* Z& V7 q( j: b/ p/ u* V - printf("in_data:");1 [ m/ f' V) M6 j3 t% o
- for (int i=0; i<AI_HAR_IGN_IN_1_SIZE; i++)
# c" R8 D M% S) t! I) ^& T - {
' {0 M8 P4 W! ]* P6 K, z0 k - switch(i%3){+ C0 w# ^. ?, W) t
- case 0:
g" x; X# ?1 F8 H6 V$ ?$ v - ((ai_float*)in_data)[i] = -175+(ai_float)(i*factor*1.2)/10.0;: ]: W; i, ~6 o
- break;
# u0 d! y. B9 X) A. W$ g& s# C - case 1:
1 p( c- S3 e5 x7 C6 ], `) O% Q - ((ai_float*)in_data)[i] = 50+(ai_float)(i*factor*0.6)/100.0;
( l$ q& T0 I3 ~3 _# U- ^ - break;
/ _( y5 M4 d9 K2 K/ T- J( S - case 2:
5 U( @& ~+ o% B4 G0 L- } - ((ai_float*)in_data)[i] = 975-(ai_float)(i*factor*1.8)/100.0;' x0 S9 m0 x# y( h1 I7 i; e5 B
- break;
0 n W; a4 d- b' D( o; s - default:: L/ C2 t6 f. q# {* D0 Z3 l! r, L% v
- break;
4 o6 w+ J7 z0 d& W3 s; ^2 g8 G$ L - }8 i# P/ C# A# @
- printf("%f ",((ai_float*)in_data)[i]);6 {. }) E- q! N3 a
- }9 q2 x! M j0 N
- printf("\n");' J; v; t- R2 {
- return 0;
+ t `! N+ V, p3 ^) B. q - }4 e0 E0 S& |% c- i3 \; T4 m w( s
- /*
8 Q/ W" F+ B* N - * Run inference1 H# \/ x6 K3 t$ _& E6 g5 Z
- */
. L4 d8 Q h# J7 T. x# V+ J - int aiRun(const void *in_data, void *out_data) {
8 [6 {" ~* c1 l6 {& b7 D0 V - ai_i32 n_batch;# f+ B8 x F( Y2 e
- ai_error err;
i+ c8 X0 o' i( [- A; D - . s, l4 D$ D5 M% r: Y
- /* 1 - Update IO handlers with the data payload */
2 f8 r2 ?0 p! ?# @ - ai_input[0].data = AI_HANDLE_PTR(in_data);
% {) h2 n3 M5 ]. o6 d! U" L9 E$ ` - ai_output[0].data = AI_HANDLE_PTR(out_data);
: S- j. |: _% w6 e: Z0 m$ I -
$ H& p- _7 a8 b1 ~4 [ - /* 2 - Perform the inference */
; Q9 F; w" N5 o& E- L. p - n_batch = ai_har_ign_run(network, &ai_input[0], &ai_output[0]);
" R% @6 F3 w+ E6 b1 Z* ?# X - if (n_batch != 1) {
7 {: M4 |$ e ]1 Y - err = ai_har_ign_get_error(network);. O" r( ~; N0 h$ Y( v
- printf("ai_error_type:%d,ai_error_code:%d\r\n",err.type,err.code);
' h$ R% c1 E( `% {. K - };' r2 V" p- `+ N! E
-
# {# a* X; H; A( H - return 0;
# F; F% H0 O% E, g4 h - }( F, [1 D. ]# Z- M* I
-
8 X7 K1 M0 |2 Q& y; B* p2 p1 \ - int post_process(void *out_data)% e* a7 H* v# m
- {/ ]+ _" [) y, K. i2 o
- printf("out_data:");4 m& E4 @4 B; S) W
- for (int i=0; i<AI_HAR_IGN_OUT_1_SIZE; i++)
; U: P* x! {9 C, z j0 _ - {% ~1 Q1 `. m; v+ D" m; n! `& X
- printf("%f ",((ai_float*)out_data)[i]);& ]# z! r0 Q4 A) d" I# _; v; Q
- }% J/ o7 N1 t* w! g" A t) [
- printf("\n");) i" v3 F8 e/ C6 h
- return 0;- J7 |6 \" t- k7 ?5 d0 }$ z
- }
5 S9 J. p. n/ q! j' s - /* USER CODE END 0 */9 C0 y* s# D" l5 k9 T3 C
- h+ B" m( m0 K5 L
- /**- {, c' q% A& _
- * @brief The application entry point.
4 k" J7 n3 p6 [6 I' X( A# X - * @retval int
_3 X' K7 w0 O3 g3 v$ t/ O4 ~ - */8 w5 D% M/ |2 @2 K" L$ G9 H
- int main(void)7 O7 `2 p: d' n& W$ @4 t
- {
9 y" I# p. m7 c- c2 D; C. v - /* USER CODE BEGIN 1 */: n9 k# f- i) @0 G- O( A0 h
-
8 X% \. ?6 G% }; T4 y - /* USER CODE END 1 */' E4 v4 v. k# v6 S! L) i/ k
-
T( P; i2 |# \% b$ {+ @ - /* MCU Configuration--------------------------------------------------------*/
9 Z' H# V# t4 M2 ?% e# K6 S -
) ]% ~0 u; _7 Z7 g6 M - /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
U- Y3 Q% u# Z - HAL_Init();3 N) G- U' b9 b
- ; C* p7 V& t- M( x2 S. z" f0 o
- /* USER CODE BEGIN Init */
& V6 s% y* s2 a. Q6 \( k! D; a, [ -
1 @$ A; t, z1 O, M - /* USER CODE END Init */
9 l: A. }0 @. {! N a+ b - ) r$ U9 D2 O3 Q' o/ x! h! y/ {: f
- /* Configure the system clock */
( p* V3 m/ C8 z - SystemClock_Config();
+ ~$ x; T, s4 S; J9 j -
6 o! P, I0 M) D! R - /* USER CODE BEGIN SysInit */
! j# j) L7 R; s/ l - " K5 Q4 L( a; o! g" h4 b [7 ]
- /* USER CODE END SysInit */
6 `$ Y' U' t0 f: v8 X - % v( m' W) d* }+ G% M% d
- /* Initialize all configured peripherals */
! {+ v. x2 O9 }: G! R - MX_GPIO_Init();
+ ^9 X2 c8 w9 k1 }- g3 {& r* k2 Q* R - MX_DFSDM1_Init();
/ @* J( | v8 M7 @. g1 A8 C - MX_I2C2_Init();; f& e8 e0 j8 H+ ~) r2 i2 E f
- MX_QUADSPI_Init();0 l8 @- E% G( p# X: ^$ E* r1 c3 B
- MX_SPI3_Init();9 `4 L" T# Q" U9 k& G6 z
- MX_USART1_UART_Init();& v; S$ o7 ]6 Z# p1 L! I: W" f: |! |
- MX_USART3_UART_Init();# G3 E, q: s! I* h) E
- MX_USB_OTG_FS_PCD_Init();+ A5 _8 _8 J) w- }& y$ ?
- MX_CRC_Init();- F- w% C9 u2 q1 m
- /* USER CODE BEGIN 2 */
8 w3 H6 n q/ n4 m - ResetPrintInit(&huart1);; b1 M* g# Y+ ?) Q8 t2 Z7 l8 T8 a0 ]- h
- HAL_UART_Receive_IT(&huart1,(uint8_t *)&USART_NewData, 1); //再开启接收中断
B+ x1 `$ j$ D0 N1 M3 z; G - USART_RX_STA = 0;
( F" Z/ p) @+ x, ] ~4 C - aiInit();; ?0 A+ w; ^/ E: n6 _& v6 b# r+ N
- uint8_t factor = 1;1 | C0 o+ c% w
- buf_print();' h& q& c1 P9 H: B1 X
- /* USER CODE END 2 */- g1 N* ^) @ L5 W
- 6 \! M; k# [3 n) ?8 n
- /* Infinite loop */
8 O w; }: H/ X8 O# |7 U3 t9 R7 J - /* USER CODE BEGIN WHILE */
|0 G' t. a/ L, H: b- Z! W2 @3 E* S8 ? - while (1)
! c* K: T* \) k- |, [ - {7 D7 g: ]1 e# j; C9 W- z! |! T3 Z
- if(USART_RX_STA&0xC000){//溢出或换行,重新开始
: V7 V4 N* g- [5 j8 O2 ?; B" n& w - printf("uart1:%.*s\r\n",USART_RX_STA&0X0FFF, USART_RX_BUF);& \0 O3 ^# G1 h
- if(strstr((const char*)USART_RX_BUF,(const char*)"test"))
, _9 X' {( q4 |' Y9 n' r - {
2 j: e' `; ^/ e4 K9 R( ]& b& U4 i - factor = ((uint8_t)USART_RX_BUF[4]-0x30);
( B* k5 y2 _' w& ~' e7 \. o - printf("factor:%d\n",factor);
1 X: X6 Z/ E" Z* x. ?4 v$ \ - acquire_and_process_data(in_data,factor);
7 o' m! l5 ?: N4 a; A - aiRun(in_data, out_data);1 f8 S3 Q1 V2 B3 `' Q
- post_process(out_data);
; ?# S- B+ C( r/ ~ - }
" a4 m1 u P! X, W$ G3 R - USART_RX_STA=0;//接收错误,重新开始
1 l4 Y5 Z; E' ]( l2 t+ `' d - HAL_Delay(100);//等待
- P% c' e5 t4 ] - }
) e# v) ]* F0 y5 M3 K% a4 r - /* USER CODE END WHILE */
3 C8 w/ ?! k# f1 V7 ?) j - ' U- @+ T8 d, U" l$ Z6 D. t) ^- }; |: P
- /* USER CODE BEGIN 3 */& S( g" ? S) b! z9 l1 ^2 C
- }
% a; t& |+ u; g" a6 u - /* USER CODE END 3 */
! ]( V3 A0 m/ W% j' { - }
' e! X1 j, D9 f% } U7 j( R4 {) C1 u - //其他生产代码) r# N$ w, t* [3 V. y `+ Y# V( @; W
- .......
复制代码
4 b$ k' M D5 U+ Y 7.4 编译及程序运行测试
& H7 z. q# p7 R5 H! i. w 配置工程输出文件格式支持,并设置运行配置:+ U! V0 f7 H" d/ e/ s4 M! B
: w9 _" R2 D u" q7 Y
5 ~$ z1 L& A( G0 [) `5 o
0 [: n2 F+ g$ d7 l
编译及下载程序:
7 S2 h6 n+ I) F7 F) ]0 v; ~
/ U/ a0 B0 ?; A; @4 E- }" a
: i; y1 i2 S3 I7 P$ \( V# q
2 C) u* f3 S B1 n 打开串口助手,查看日志输出,发送信息,例如test7,即7作为因子生成输入数据,然后看输出结果。/ A- i" ]( N3 u; H0 H6 d
1 @' B7 S6 J' |5 n
2 w' C- U' W- P. f2 O: u
+ h" G/ S9 |. j" V! A; {' _
7.5 补充说明
8 {, |5 ^ ~. c( }1 X 目前只能说是采用cubeIDE+cube.AI+keras的STM32嵌入式人工智能开发走通了流程,但是串口反馈回来的日志信息是不合理的,因为在数据采集时我们只采集了传感器的三个数值,但在训练模型时,默认的数据输入量是24,显然是不合理的,因此需要还需要重新分析官方提供的HAR训练模型的项目,使得模型训练与采集数据匹配起来,请阅读篇二。- z. H' j D3 d5 q& a. ?4 {; q
, ~, q2 J. { K' }# A! K" o
4 @' i% b& j; q; @2 G3 U! {
0 P3 N) o7 V5 t2 D6 j5 M
但考虑到官方提供的HAR训练模型的工程项目还是过于复杂,不助于学习和了解cube.AI的真正用法,因此后面将抛弃官方提供的HAR训练模型的项目,自行撰写一个训练模型项目+实际采集数据生成神经网络模型,是的数据输入和输出匹配,并将采用传感器实时采集到的数据进行计算评估. g& P+ A4 a2 H& G
————————————————
! F8 ~1 |% g3 K. m9 g版权声明:py_free-物联智能
9 q6 @" T* n# U: L, b1 L如有侵权请联系删除
! g6 p' y2 R. n2 I8 L8 [9 P2 W/ U3 y' l" n+ x) y
4 d* M2 s6 t' ?% c' x; R& e
! K _( K! G( B
4 ^. f7 H% V% p- v6 C |