写的教程反响平平，可能是晦涩难懂，那就直接上代码（极度粗糙的demo）。
- G2 I5 {0 N$ n5 \- l" n  @  接上次的软硬件环境，我采用的STemwin（本来想用littlevgl）自带bmp显示（详细代码可参考某某莱教程）。先看看生成的一些代码。

+ `/ `/ e) m: p7 ?3 w                GUI_Init();
( A8 a7 A% m/ D5 p! f                GUI_SetBkColor(GUI_BLUE_98);
                GUI_Clear();
/ J8 ~( `# X  g/ k5 F) Z( B       
                GUI_DispStringHCenterAt( GUI_GetVersionString() , GUI_GetScreenSizeX()/2, GUI_GetScreenSizeY()/2);
                SDRes=f_mount(&SDFatFS,(TCHAR*)SDPath,0);
                if(SDRes!=FR_OK)
+ E7 v. k8 y9 P2 q3 @                {
4 m' l5 u9 F" l/ B. ^                Error_Handler();
) F; l% F9 n1 a' k) n& m. X                }
) G$ n4 l' J5 {& V2 x0 D  MX_X_CUBE_AI_Process();
  /* USER CODE END 2 */
  /* Infinite loop */
  /* USER CODE BEGIN WHILE */
  while (1)
" \9 o; _" O  E0 K' m! @6 H2 @  {
    /* USER CODE END WHILE */
: s$ U- i9 W. O, ]# r. a                         GUI_Delay(10);
    /* USER CODE BEGIN 3 */
  }

* q- P! \/ R, x: e, i, J4 Z简单的主函数。
5 l: g: e7 j! P4 c* w

$ X( q. m* h3 B( I- F  进入MX_X_CUBE_AI_Process();
$ l- c) l4 R3 W' P0 L

    ai_buffer ai_input;
    ai_buffer ai_output;
, g" L8 J- w6 M/ ?" F% N3 A+ B: _. ~    ai_i32 batch;
    char str_tmp[16];
     for(uint8_t i=0; i<10; i++)
            {
                                                sprintf(str_tmp,"%1d.bmp",i);
; |+ a4 Z1 O5 X- e( s1 [                                          _ShowBMPEx(str_tmp);
                                                LCD_ShowBMP(str_tmp);
                                                for(uint16_t i=0; i<AI_LENET_IN_1_SIZE ; i++ )
                                                {
                                                                in_data[i] = *bmpBuffPtr++;
                                                }
                                                ai_input.n_batches  = 1;
                                                ai_input.data = AI_HANDLE_PTR(in_data);
( u. }) m( u$ [. l' G                                                ai_output.n_batches = 1;
- A& u* ^1 m0 t                                                ai_output.data = AI_HANDLE_PTR(out_data);
. ^; V. Q' [. h4 G" c' f, O3 S: T                                                batch = ai_lenet_run(handel, &ai_input, &ai_output);
3 W6 e- ]- N- C! \0 x: q                                                if (batch != 1)
                                                {
                                                                err = ai_lenet_get_error(handel);
                                                                if( err.code )
                                                                {
* a" i+ ^  _$ r( n% [  F                                                                                while(1);
                                                                }
                                                }
                                         uint8_t count=0;
7 p  S7 v: I, _0 M* J/ H" y3 a                                         for(uint8_t i=0; i<AI_MNETWORK_OUT_1_SIZE; i++)
% H: g4 Q# U  |9 S( d4 S4 V                                                {
                                                  if( out_data[i] >=0.9f)
+ W  p4 A) G* t7 Y3 C( ]                                                                {
                                                                         sprintf(str_tmp,"Num:%d",i);
2 ~  S# |& u& l; s$ F" U                                                                         GUI_SetFont(GUI_FONT_32B_ASCII);
                                                                  GUI_DispStringAt(str_tmp,200,0);
                                                                         GUI_SetFont(GUI_FONT_16B_ASCII);
                                                                         count++;
                                                                }
                                                }
- q, L6 f. E7 @      if(count!=1)
" e1 ?. E3 M4 H" X6 F; ]1 C9 c* u$ A                                                {
                                                        GUI_SetFont(GUI_FONT_32B_ASCII);
                                                        GUI_DispStringAt("err",200,0);
                                                        GUI_SetFont(GUI_FONT_16B_ASCII);
                                                }
                                                free(bmpBuffPtr);
& S& k1 u- F# U: q                                                HAL_Delay(2000);
                                        }

7 d9 D+ ~6 O6 W      熟悉STemwin的大佬肯定知道我在干嘛，不断的读取SD卡中的bmp文件，然后显示，并且通过ai_lenet_run函数得到识别结果，输出是static ai_float out_data[AI_MNETWORK_OUT_1_SIZE]; 0-9有十个识别结果，这个向量大小是十。调试程序时，输出不像是我在PC ubuntu caffe环境中的输出结果，PC上主要是以softmax得到每个类别的置信度，所以PC上向量肯定全部都有值，置信度最高即为模型结果。在MCU上，输出结果只有零和一。让我很奇怪，既然是ai_float 型 何必不输出 所有的置信度。后面我将继续挖掘。
- A6 k9 z5 a2 x  d+ T. Y
       流程介绍完毕，下面说一说cubeAI的主要函数。
# y7 ~1 q( E0 ~       ai_lenet_create(&handel, NULL);
/ i1 _; x" I) [! g# Z/ F; M: o( M% Z      ai_lenet_get_info(handel, &report);
! K3 I( H: v6 l& S) ^. L% j          ai_network_params params = {
5 p* n6 l# v( O) X2 `" r                                                          AI_LENET_DATA_WEIGHTS(ai_lenet_data_weights_get()),    //权重的获取
                                                        AI_LENET_DATA_ACTIVATIONS(activations)     // 激活函数
5 v9 A. {. c' c! J- ^8 W1 \                                                };
' j4 Z& K; t2 S( _2 a- s          ai_lenet_init(handel, &params);  //lenet的初始化，估计是申请内存什么的，写flash。
                                                                         err = ai_lenet_get_error(handel);
  E" A! v2 Y. R' k3 y+ S0 P& b
$ J1 {6 {4 b7 t; j! Z* M         batch = ai_lenet_run(handel, &ai_input, &ai_output); //图片输入以及处理完毕的输出，1batch批

" a3 h5 s- l  k) ^/ p% f) V, I
         总的来说比较简单的demo，其实也可以做复杂一点，先手写然后STemwin存bmp图片，然后利用加权平均得到BMP颜色数据的灰度，输入到lenet，dedaojieguo。效果会比较酷炫。
        这模型被压缩很厉害，能有这样的结果以及不错了，最后祝贺一下ST，半导体前15！  期待MP1，玩玩不用压缩的AI。
        附件：工程源码。


( X% e  P5 a) ~# ]

兄弟，您这个代码运行最后的准确率能达到多少？我今天试了一下，10个数有9个识别错误的。

支持下

strang 发表于 2019-5-18 09:02
支持下

- s9 k- M" @7 Y7 h+ s谢谢支持

支持支持

支持下         

加油楼主，我也在研究这个CUBE AI，进度没有你快

学习学习!