mirror of
https://github.com/mit-han-lab/tinyengine.git
synced 2025-05-09 17:11:01 +08:00
34 lines
1.3 KiB
C
34 lines
1.3 KiB
C
/* ----------------------------------------------------------------------
|
|
* Project: TinyEngine
|
|
* Title: yoloOutput.h
|
|
*
|
|
* Reference papers:
|
|
* - MCUNet: Tiny Deep Learning on IoT Device, NeurIPS 2020
|
|
* - MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning, NeurIPS 2021
|
|
* - MCUNetV3: On-Device Training Under 256KB Memory, NeurIPS 2022
|
|
* Contact authors:
|
|
* - Wei-Ming Chen, wmchen@mit.edu
|
|
* - Wei-Chen Wang, wweichen@mit.edu
|
|
* - Ji Lin, jilin@mit.edu
|
|
* - Ligeng Zhu, ligeng@mit.edu
|
|
* - Song Han, songhan@mit.edu
|
|
*
|
|
* Target ISA: ARMv7E-M
|
|
* -------------------------------------------------------------------- */
|
|
|
|
typedef struct box{
|
|
float x0;
|
|
float y0;
|
|
float x1;
|
|
float y1;
|
|
float score;
|
|
} det_box;
|
|
|
|
det_box** postprocessing(signed char *input_data[3], signed char y_zero[3], float y_scale[3],
|
|
unsigned char *data_buf, int w, int h, int output_c, int num_classes, const int anchors[3][3][2], int outputs,
|
|
const float NMS_threshold, const float VALID_THRESHOLD, int* box_ret, det_box** ret_box);
|
|
|
|
det_box** postprocessing_fp(float *input_data[3], signed char y_zero[3], float y_scale[3],
|
|
unsigned char *data_buf, int w, int h, int output_c, int num_classes, const int anchors[3][3][2], int outputs,
|
|
const float NMS_threshold, const float VALID_THRESHOLD, int* box_ret, det_box** ret_box);
|