#ifndef _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_
#define _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_

#include <stdint.h>
#include <vector>
#include "rknn_api.h"
#include "PPYOLOE.hpp"

// class rknn_app_context_t;

typedef struct
{
    int left;
    int top;
    int right;
    int bottom;
} image_rect_t;

typedef struct
{
    image_rect_t box;
    float prop;
    int cls_id;
} object_detect_result;

typedef struct
{
    int id;
    int count;
    object_detect_result results[128];
} object_detect_result_list;

int init_post_process();
void deinit_post_process();
char *coco_cls_to_name(int cls_id);
int post_process(PPYOLOE *app_ctx, rknn_output *outputs, float conf_threshold, float nms_threshold, object_detect_result_list *od_results, int OBJ_CLASS_NUM);

void deinitPostProcess();
#endif //_RKNN_PPYOLOE_DEMO_POSTPROCESS_H_