ai.h 716 B

12345678910111213141516171819202122232425262728293031323334353637
  1. /*
  2. * ai.h
  3. *
  4. * Created on: Aug 25, 2022
  5. * Author: pgj
  6. */
  7. #ifndef INC_AI_H_
  8. #define INC_AI_H_
  9. // Expose a C friendly interface for main functions.
  10. #ifdef __cplusplus
  11. extern "C" {
  12. #endif
  13. #include "stm32l496xx.h"
  14. //#define input_size 32*32*3
  15. //#define state_size 4
  16. //extern float input_image[input_size];
  17. //extern float state[state_size];
  18. //extern char float_data[30];
  19. //
  20. //const float model_energy_req[5] = {0.550483672, 0.220868206, 0.186921805, 0.036269573, 0.02993864};
  21. uint8_t* setup(uint8_t *, int, float *scale, int32_t* zero_point);
  22. void preprocess_input();
  23. int perform_inference();
  24. void print_output();
  25. #ifdef __cplusplus
  26. }
  27. #endif
  28. #endif /* INC_AI_H_ */