detectionUtility.h 1.0 KB

123456789101112131415161718192021222324252627
  1. /* ----------------------------------------------------------------------
  2. * Project: TinyEngine
  3. * Title: detectionUtility.h
  4. *
  5. * Reference papers:
  6. * - MCUNet: Tiny Deep Learning on IoT Device, NeurIPS 2020
  7. * - MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning, NeurIPS 2021
  8. * - MCUNetV3: On-Device Training Under 256KB Memory, NeurIPS 2022
  9. * Contact authors:
  10. * - Wei-Ming Chen, wmchen@mit.edu
  11. * - Wei-Chen Wang, wweichen@mit.edu
  12. * - Ji Lin, jilin@mit.edu
  13. * - Ligeng Zhu, ligeng@mit.edu
  14. * - Song Han, songhan@mit.edu
  15. *
  16. * Target ISA: ARMv7E-M
  17. * -------------------------------------------------------------------- */
  18. #ifndef TINYENGINE_INCLUDE_DETECTIONUTILITY_H_
  19. #define TINYENGINE_INCLUDE_DETECTIONUTILITY_H_
  20. int postProcessing(signed char *input, unsigned char* runtime_buffer,
  21. int y_zero, float y_scale, int shape_x, int shape_y, int shape_c, int resolution,
  22. int width, int height , float conf_thresh, float out_boxes[10][6]);
  23. #endif /* TINYENGINE_INCLUDE_DETECTIONUTILITY_H_ */