mirror of
https://github.com/azaion/autopilot.git
synced 2026-04-22 22:06:34 +00:00
Tuomas Järvinen
37 lines
909 B
C++
37 lines
909 B
C++
#ifndef _RKNN_YOLOV8_DEMO_POSTPROCESS_H_
|
|
#define _RKNN_YOLOV8_DEMO_POSTPROCESS_H_
|
|
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
#include "rknn_api.h"
|
|
#include "common.h"
|
|
#include "image_utils.h"
|
|
|
|
#define OBJ_NAME_MAX_SIZE 64
|
|
#define OBJ_NUMB_MAX_SIZE 128
|
|
#define OBJ_CLASS_NUM 80
|
|
#define NMS_THRESH 0.45
|
|
#define BOX_THRESH 0.25
|
|
|
|
// class rknn_app_context_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[OBJ_NUMB_MAX_SIZE];
|
|
} object_detect_result_list;
|
|
|
|
int init_post_process();
|
|
void deinit_post_process();
|
|
char *coco_cls_to_name(int cls_id);
|
|
int post_process(rknn_app_context_t *app_ctx, rknn_output *outputs, letterbox_t *letter_box, float conf_threshold, float nms_threshold, object_detect_result_list *od_results);
|
|
|
|
void deinitPostProcess();
|
|
#endif //_RKNN_YOLOV8_DEMO_POSTPROCESS_H_
|