postprocess.h 542 B

123456789101112131415161718
  1. #ifndef _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_
  2. #define _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_
  3. #include <stdint.h>
  4. #include <vector>
  5. #include "rknn_api.h"
  6. #include "PPYOLOE.hpp"
  7. #include "inferResult.h"
  8. // class rknn_app_context_t;
  9. int init_post_process();
  10. void deinit_post_process();
  11. char *coco_cls_to_name(int cls_id);
  12. 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);
  13. void deinitPostProcess();
  14. #endif //_RKNN_PPYOLOE_DEMO_POSTPROCESS_H_