first commit

2025-08-22 13:52:24 +08:00
commit 05a551d523
257 changed files with 49217 additions and 0 deletions
@@ -0,0 +1,33 @@
+#ifndef _RKNN_YOLOV5_DEMO_POSTPROCESS_H_
+#define _RKNN_YOLOV5_DEMO_POSTPROCESS_H_
+
+#include <stdint.h>
+#include <vector>
+#include "rknn_api.h"
+#include "common.h"
+#include "yolov5.h"
+
+#define RKNPU1
+
+// #define PROP_BOX_SIZE (5 + OBJ_CLASS_NUM)
+
+extern int OBJ_CLASS_REAL_NUM;
+extern int PROP_BOX_SIZE;
+
+extern char *labels[OBJ_MAX_CLASS_NUM];
+extern int codes[OBJ_MAX_CLASS_NUM];
+
+// class rknn_app_context_t;
+typedef struct {
+    int x_pad;
+    int y_pad;
+    float scale;
+} letterbox_t;
+
+int init_post_process();
+void deinit_post_process();
+const char *coco_cls_to_name(int cls_id);
+int post_process(rknn_app_context_t *app_ctx, void *outputs, letterbox_t *letter_box, float conf_threshold, float nms_threshold, object_detect_result_list *od_results);
+
+void deinitPostProcess();
+#endif //_RKNN_YOLOV5_DEMO_POSTPROCESS_H_