GK SDK 源码库: XMIPCLinuxV100R005C00SPC030 (kernel/tools/open_source excluded)

source/gmp/usr/svp/include/task_instance/xm_adas.h

@@ -0,0 +1,54 @@
+#ifndef __XM_ADAS_H__
+#define __XM_ADAS_H__
+
+#include "svp_comm.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#define SVP_LANE_ROW_SCORE_NUM 100
+#define SVP_LANE_COL_SCORE_NUM 50
+#define SVP_LANE_INPUT_IMG_WIGTH 800
+#define SVP_LANE_INPUT_IMG_HIGHT 532
+#define SVP_LANE_CROP_IMG_HIGHT 320
+#define SVP_PLATE_IMG_WIGTH 512
+#define SVP_PLATE_IMG_HIGHT 288
+#define SVP_FIT_ORDER 3
+#define SVP_FIT_MAX_RESIDUALS 15
+#define SVP_FIT_MIN_NUM 7
+
+typedef struct {
+    xmedia_npu_model npu_model;
+} svp_lane_info;
+
+typedef struct {
+    svp_lane_info *lane_info;
+    yolov5_detect_param *plate_detect_info;
+    yolov5_detect_param *car_detect_info;
+    yolov5_detect_param *person_detect_info;
+    yolov5_detect_param *nmv_detect_info;
+    xmedia_svp_adas_distance_attr dist_attr;
+} svp_adas_information;
+
+typedef struct {
+    svp_adas_information info;
+    xmedia_void *priv;
+} svp_adas_context;
+
+xmedia_s32 adas_init(xmedia_void *context, const xmedia_svp_task_cfg cfg);
+
+xmedia_s32 adas_uninit(xmedia_void *context);
+
+xmedia_s32 adas_set_attr(xmedia_void *context, const xmedia_svp_adas_attr *task_attr);
+
+xmedia_s32 adas_get_attr(xmedia_void *context, xmedia_svp_adas_attr *task_attr);
+
+xmedia_s32 adas_process(xmedia_void *context, const xmedia_svp_task_input *input,
+                            xmedia_svp_adas_result *output);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif