[STM32U5] 【NUCLEO-U575ZI-Q测评】使用轻量级AI推理框架TinyMaix实现手写数字识别

/******************************* MODEL FUNCTION ************************************/

tm_err_t tm_load  (tm_mdl_t* mdl, const uint8_t* bin, uint8_t*buf, tm_cb_t cb, tm_mat_t* in);   //load model

void     tm_unload(tm_mdl_t* mdl);                                      //remove model

tm_err_t tm_preprocess(tm_mdl_t* mdl, tm_pp_t pp_type, tm_mat_t* in, tm_mat_t* out);            //preprocess input data

tm_err_t tm_run   (tm_mdl_t* mdl, tm_mat_t* in, tm_mat_t* out);         //run model

​

/******************************* UTILS FUNCTION ************************************/

uint8_t TM_WEAK tm_fp32to8(float fp32);

float TM_WEAK tm_fp8to32(uint8_t fp8);

​

/******************************* STAT FUNCTION ************************************/

#if TM_ENABLE_STAT

tm_err_t tm_stat(tm_mdlbin_t* mdl);                    //stat model

#endif

/******************************* LAYER FUNCTION ************************************/

tm_err_t tml_conv2d_dwconv2d(tm_mat_t* in, tm_mat_t* out, wtype_t* w, btype_t* b, \

    int kw, int kh, int sx, int sy, int dx, int dy, int act, \

    int pad_top, int pad_bottom, int pad_left, int pad_right, int dmul, \

    sctype_t* ws, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp);

tm_err_t tml_gap(tm_mat_t* in, tm_mat_t* out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp);

tm_err_t tml_fc(tm_mat_t* in, tm_mat_t* out,  wtype_t* w, btype_t* b, \

    sctype_t* ws, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp);

tm_err_t tml_softmax(tm_mat_t* in, tm_mat_t* out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp);

tm_err_t tml_reshape(tm_mat_t* in, tm_mat_t* out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp);

tm_err_t tml_add(tm_mat_t* in0, tm_mat_t* in1, tm_mat_t* out, \

    sctype_t in_s0, zptype_t in_zp0, sctype_t in_s1, zptype_t in_zp1, sctype_t out_s, zptype_t out_zp);

#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

#define  TM_GET_US()       ((uint32_t)((uint64_t)clock()*1000000/CLOCKS_PER_SEC))

​

#define TM_DBGT_INIT()     uint32_t _start,_finish;float _time;_start=TM_GET_US();

#define TM_DBGT_START()    _start=TM_GET_US();

#define TM_DBGT(x)         {_finish=TM_GET_US();\

                            _time = (float)(_finish-_start)/1000.0;\

                            TM_PRINTF("===%s use %.3f ms\n", (x), _time);\

                            _start=TM_GET_US();}

int fputc(int ch, FILE* f)

{

  (void) f;

​

  uint8_t data[] = {ch};

  if (HAL_UART_Transmit(&huart1, data, sizeof(data), 1000) != HAL_OK)

  {

    return EOF;

  }

​

  return ch;

}

  /* Infinite loop */

  /* USER CODE BEGIN WHILE */

  while (1)

  {

    HAL_GPIO_WritePin(LED_RED_GPIO_Port, LED_RED_Pin, GPIO_PIN_RESET);

    HAL_Delay(1000);

    printf("Low\r\n");

​

    HAL_GPIO_WritePin(LED_RED_GPIO_Port, LED_RED_Pin, GPIO_PIN_SET);

    HAL_Delay(1000);

    printf("High\r\n");

    /* USER CODE END WHILE */

​

    /* USER CODE BEGIN 3 */

  }

  /* USER CODE END 3 */

#include "stm32u5xx_hal.h"

#define TM_GET_MS()         HAL_GetTick()

#define TM_DBGT_INIT()      uint32_t _start, _finish; int _time; _start = TM_GET_MS();

#define TM_DBGT_START()     _start = TM_GET_MS();

#define TM_DBGT(x)          _finish = TM_GET_MS(); \

                            _time = (_finish - _start); \

                            TM_PRINTF("===%s use %d ms\n", (x), _time); \

                            _start = TM_GET_MS();

  int mnist_main(int argc, char** argv);

  mnist_main(0, 0);

#if TM_MDL_TYPE == TM_MDL_INT8

#include "../../tools/tmdl/mnist_valid_q.h"

//#include "../../tools/tmdl/mnist_resnet_q.h"

#elif TM_MDL_TYPE == TM_MDL_FP32

#include "../../tools/tmdl/mnist_valid_f.h"

//#include "../../tools/tmdl/mnist_resnet_f.h"

#elif TM_MDL_TYPE == TM_MDL_FP16

#include "../../tools/tmdl/mnist_valid_fp16.h"

#elif TM_MDL_TYPE == TM_MDL_FP8_143

#include "../../tools/tmdl/mnist_fp8_143.h"

#elif TM_MDL_TYPE == TM_MDL_FP8_152

#include "../../tools/tmdl/mnist_fp8_152.h"

#endif