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

2026-09-06 03:52:57 +08:00
commit b1928b41c0
21813 changed files with 4413081 additions and 0 deletions
@@ -0,0 +1,72 @@
+#ifndef __XM_DMS_H__
+#define __XM_DMS_H__
+
+#include "svp_comm.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+#define DMS_EPSILON 1e-6f
+#define DMS_DETECTION_MODEL 3
+#define DMS_KEYPOINT 68
+#define DMS_INT_FRAME 10
+#define DMS_EYES_THRE 0.25
+#define DMS_OUT_MOUTH_THRES 1.3
+#define DMS_IN_MOUTH_THRES 0.7
+#define DMS_IOU_THRES 0.1
+
+typedef struct {
+    xmedia_bool close_eyes;
+    xmedia_bool yawn;
+    xmedia_bool phone;
+    xmedia_bool cigar;
+} svp_dms_flag;
+
+typedef struct {
+    xmedia_u8 close_eyes_frame;
+    xmedia_u8 yawn_frame;
+    xmedia_u8 phone_frame;
+    xmedia_u8 cigar_frame;
+    xmedia_u8 id_frame;
+} svp_dms_frame;
+
+typedef struct {
+    xmedia_float left_ear;                           // 左眼宽高比
+    xmedia_float right_ear;                          // 右眼宽高比
+    xmedia_float out_mar;                            // 外嘴宽高比
+    xmedia_float in_mar;                             // 内嘴宽高比
+    xmedia_float pitch;                              // 俯仰角
+    xmedia_float yaw;                                // 偏航角
+    xmedia_float roll;                               // 翻滚角
+} svp_dms_state;
+
+typedef struct {
+    svp_dms_flag flag;
+    svp_dms_frame frame;
+    svp_dms_state state;
+    xmedia_s32 id;
+} svp_dms_alarm_report;
+
+typedef struct {
+    yolov8_detect_param *detect_info;
+    svp_dms_alarm_report alarm_report;
+    xmedia_u8 warn_frame;                           // 警告帧数
+    xmedia_float iou_thres;                         // 相交阈值，建议值0.1f
+} svp_dms_context;
+
+xmedia_s32 dms_init(xmedia_void *context, const xmedia_svp_task_cfg cfg);
+
+xmedia_s32 dms_uninit(xmedia_void *context);
+
+xmedia_s32 dms_set_attr(xmedia_void *context, const xmedia_svp_dms_attr *task_attr);
+
+xmedia_s32 dms_get_attr(xmedia_void *context, xmedia_svp_dms_attr *task_attr);
+
+xmedia_s32 dms_process(xmedia_void *context, const xmedia_svp_task_input *input, xmedia_svp_dms_output *output);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif
+