STM32で人物検出する推論ネットワークの実装

int acquire_and_process_data(uint8_t *data)
{
  uint32_t sTime;
  uint32_t eTime;
  uint8_t eMsg[256] = {0};
  extern Image_t cameraImg;
  extern Image_t resizedImg;
  extern Image_t inputImg;

  sTime = HAL_GetTick();

  captureBMP565();

  eTime = HAL_GetTick();

  sprintf(eMsg, " Done : %lums\r\n", eTime-sTime);
  HAL_UART_Transmit(&huart2, (uint8_t *)eMsg, sizeof(eMsg), 0xFFFF);

  if ((cameraImg.width != inputImg.width) || (cameraImg.height != inputImg.height))
  {
    ImgResize(&cameraImg, &resizedImg, NEAREST);
    ImgToGrayscale(&resizedImg, &inputImg);
  }
  else
  {
    ImgToGrayscale(&cameraImg, &inputImg);
  }
  memcpy(data, inputImg.pData, INPUT_IMG_HEIGHT*INPUT_IMG_WIDTH);

  return 0;
}

static int ai_run(void *data_in, void *data_out)
{
  ai_i32 batch;

  ai_buffer *ai_input = network_info.inputs;
  ai_buffer *ai_output = network_info.outputs;

  ai_input[0].data = AI_HANDLE_PTR(data_in);
  ai_output[0].data = AI_HANDLE_PTR(data_out);

  batch = ai_network_run(network, ai_input, ai_output);
  if (batch != 1) {
    ai_log_err(ai_network_get_error(network),
        "ai_network_run");
    return -1;
  }

  return 0;
}

int post_process(uint8_t *data)
{
  uint32_t sTime;
  uint32_t eTime;
  uint8_t eMsg[256] = {0};
  extern Image_t inputImg;
  extern Image_t cameraImg;

#if defined(USE_PMODMTDS)
  sTime = HAL_GetTick();

  displayReInit();
  displayImage(&cameraImg, 120-cameraImg.height/2, 160-cameraImg.width/2);

  eTime = HAL_GetTick();
  sprintf(eMsg, "Done : %lums\r\n", eTime-sTime);
  HAL_UART_Transmit(&huart2, (uint8_t *)eMsg, sizeof(eMsg), 0xFFFF);
#endif

  if ((data[1] > data[2])) {
    char msg[] = "\r\n______ Person Detected!!\r\n\r\n";
    HAL_UART_Transmit(&huart2, (uint8_t *)msg, sizeof(msg), 0xFFFF);
    HAL_GPIO_WritePin(GPIOB, GPIO_PIN_10, GPIO_PIN_SET);
#if defined(USE_BUZZER)
    myPWMStart(10, 1000, 500);
    HAL_Delay(100);
    myPWMStop();
    HAL_Delay(100);
    myPWMStart(10, 1000, 500);
    HAL_Delay(100);
    myPWMStop();
#endif
#if defined(USE_PMODMTDS)
    displayPersonDetected();
#endif
  } else {
    char msg[] = "\r\n______ No Person\r\n\r\n";
    HAL_UART_Transmit(&huart2, (uint8_t *)msg, sizeof(msg), 0xFFFF);
    HAL_GPIO_WritePin(GPIOB, GPIO_PIN_10, GPIO_PIN_RESET);
#if defined(USE_BUZZER)
    myPWMStop();
#endif
#if defined(USE_PMODMTDS)
    displayNoPersonDetected();
#endif
  }

  return 0;
}

---