| 123456789101112131415161718 |
- #ifndef _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_
- #define _RKNN_PPYOLOE_DEMO_POSTPROCESS_H_
- #include <stdint.h>
- #include <vector>
- #include "rknn_api.h"
- #include "PPYOLOE.hpp"
- #include "inferResult.h"
- // class rknn_app_context_t;
- 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_
|
