#ifndef __XMEDIA_SVP_H__ #define __XMEDIA_SVP_H__ #include "xmedia_type.h" #include "xmedia_video_common.h" #ifdef __cplusplus extern "C" { #endif #define XMEDIA_SVP_MAX_TASK_NUM (16u) // 支持最多任务个数 #define XMEDIA_SVP_MAX_TARGET_NUM (50u) // 单帧最多输出的目标个数 #define MAX_DETECT_KEYPOINT_NUM (68u) // yolov8关键点数量 #define XMEDIA_SVP_DMS_MAX_TARGET_NUM (5u) // DMS单帧输出目标最大个数 #define XMEDIA_SVP_DMS_SPECIAL_MAX_NUM (3u) // DMS手机和香烟最多输出的目标个数 #define XMEDIA_SVP_LANE_MAX_NUM 4 // 车道线最大数量 #define XMEDIA_SVP_LANE_AXIS_NUM 2 // 车道线方向数量 #define XMEDIA_SVP_LANE_MAX_ROW_POINT 18 // 车道线行方向关键点数量 #define XMEDIA_SVP_LANE_MAX_COL_POINT 40 // 车道线列方向关键点数量 #define XMEDIA_SVP_LANE_FIT_ORDER 3 // 车道线拟合因子数量 #define XMEDIA_SVP_FEATURE_COL 512 // 人脸特征向量纬度 #define XMEDIA_SVP_FR_KEYPOINT_NUM 5 // 人脸关键点数量 #define XMEDIA_SVP_MAX_VENC_CHN_NUM 16 // venc最大通道数 #define XMEDIA_SVP_MAX_VI_PIPE_NUM 8 // vi最大通道数 #define XMEDIA_SVP_PLATE_CHAR_MAX_NUM 21 // 车牌字符最大数量 #define XMEDIA_SVP_MAX_REGION 5 // 区域最大数量 #define XMEDIA_SVP_MAX_EDGE 10 // 区域最大边长 #define XMEDIA_SVP_PLATE_VEH_MAX_TARGET_NUM (10u) // 车牌车辆单帧输出目标最大个数 #define XMEDIA_SVP_PETBOT_MAX_TARGET 5 // 宠物机器人最大输出目标数量 // 单模型算法类别 typedef enum { XMEDIA_SVP_ALG_TYPE_PERSON, // 人形检测 XMEDIA_SVP_ALG_TYPE_FACE, // 人脸检测 XMEDIA_SVP_ALG_TYPE_CAR, // 车形检测 XMEDIA_SVP_ALG_TYPE_PET, // 宠物检测 XMEDIA_SVP_ALG_TYPE_HEAD, // 人头检测 XMEDIA_SVP_ALG_TYPE_ELECTRIC_BICYCLE, // 电瓶车检测 XMEDIA_SVP_ALG_TYPE_NON_MOTORIZED_VEHICLE, // 非机动车检测 XMEDIA_SVP_ALG_TYPE_FACE_RECOGNITION, // 人脸识别 XMEDIA_SVP_ALG_TYPE_PERSON_KEYPOINT, // 人体关键点检测 XMEDIA_SVP_ALG_TYPE_FACE_ATTRIBUTE, // 人脸属性 XMEDIA_SVP_ALG_TYPE_GESTURE, // 手势识别 XMEDIA_SVP_ALG_TYPE_BOUNDARY_PERSON, // 周界检测(人) XMEDIA_SVP_ALG_TYPE_BOUNDARY_CAR, // 周界检测(车) XMEDIA_SVP_ALG_TYPE_OFF_DUTY, // 离岗检测 XMEDIA_SVP_ALG_TYPE_PFS, // 客流量检测 XMEDIA_SVP_ALG_TYPE_FIREWORKS, // 烟火检测 XMEDIA_SVP_ALG_TYPE_PACKAGE, // 包裹检测 XMEDIA_SVP_ALG_TYPE_FIRESMOKE, // 烟火检测YOLOV8 XMEDIA_SVP_ALG_TYPE_SECOND_STAGE, // 二阶段检测 XMEDIA_SVP_ALG_TYPE_PERSON_CAR, // 人车检测 XMEDIA_SVP_ALG_TYPE_DMS, // DMS检测 XMEDIA_SVP_ALG_TYPE_LICENSE_DETECT, // 车牌检测 XMEDIA_SVP_ALG_TYPE_LICENSE_RECO, // 车牌字符识别 XMEDIA_SVP_ALG_TYPE_LICENSE_COLOR, // 车牌颜色识别 XMEDIA_SVP_ALG_TYPE_VEHICLE_DETECT, // 车辆检测 XMEDIA_SVP_ALG_TYPE_VEHICLE_RECO, // 车辆识别 XMEDIA_SVP_ALG_TYPE_PCNMV_FIRE, // 人车非火 XMEDIA_SVP_ALG_TYPE_MAX, } xmedia_svp_alg_type; // 输出某个目标的类别 typedef enum { XMEDIA_SVP_CLASS_TYPE_PERSON, // 人形类别 XMEDIA_SVP_CLASS_TYPE_FACE, // 人脸类别 XMEDIA_SVP_CLASS_TYPE_CAR, // 车形类别 XMEDIA_SVP_CLASS_TYPE_PET, // 宠物类别 XMEDIA_SVP_CLASS_TYPE_HEAD, // 人头类别 XMEDIA_SVP_CLASS_TYPE_ELECTRIC_BICYCLE, // 电瓶车类别 XMEDIA_SVP_CLASS_TYPE_MASK, // 口罩类别 XMEDIA_SVP_CLASS_TYPE_PERSON_KEYPOINT, // 人体关键点类别 XMEDIA_SVP_CLASS_TYPE_BIKE, // 自行车 XMEDIA_SVP_CLASS_TYPE_BIKER, // 骑自行车车的人 XMEDIA_SVP_CLASS_TYPE_MOTOR, // 摩托车 XMEDIA_SVP_CLASS_TYPE_MOTORER, // 骑摩托车的人 XMEDIA_SVP_CLASS_TYPE_TRICYCLE, // 三轮车 XMEDIA_SVP_CLASS_TYPE_TRICYCLER, // 骑三轮车的人 XMEDIA_SVP_CLASS_TYPE_FIREWORKS_FIRE, // 火 XMEDIA_SVP_CLASS_TYPE_FIREWORKS_SMOKE, // 烟 XMEDIA_SVP_CLASS_TYPE_PACKAGE, // 包裹 XMEDIA_SVP_CLASS_TYPE_FIRE, // 火YOLOV8 XMEDIA_SVP_CLASS_TYPE_SMOKE, // 烟YOLOV8 XMEDIA_SVP_CLASS_TYPE_GESTURE_V, // 手势剪刀手 XMEDIA_SVP_CLASS_TYPE_GESTURE_THUMB, // 手势大拇指 XMEDIA_SVP_CLASS_TYPE_GESTURE_OK, // 手势OK XMEDIA_SVP_CLASS_TYPE_GESTURE_PALM, // 手势手掌 XMEDIA_SVP_CLASS_TYPE_NOGESTURE, // 无手势 XMEDIA_SVP_CLASS_TYPE_LICENSE, // 车牌类别 XMEDIA_SVP_CLASS_TYPE_MAX, } xmedia_svp_class_type; typedef enum { XMEDIA_SVP_INPUTDATA_FORMAT_YUV420SP, // 输入图像格式YUV420SP XMEDIA_SVP_INPUTDATA_FORMAT_RGB888, // 输入图像格式RGB888 XMEDIA_SVP_INPUTDATA_FORMAT_MAX, } xmedia_svp_inputdata_format; typedef enum { XMEDIA_SVP_MODEL_FILE, // 模型文件输入 XMEDIA_SVP_MODEL_MEM // 模型内存输入 } xmedia_svp_model_type; // 目标的矩阵坐标点(坐标原点为图像左上角) typedef struct { xmedia_float x1; xmedia_float y1; xmedia_float x2; xmedia_float y2; } xmedia_svp_rect; typedef enum { XMEDIA_SVP_MOTION_STATE_UNKNOWN = 0, // 未知状态 XMEDIA_SVP_MOTION_STATE_STATIC, // 静止 XMEDIA_SVP_MOTION_STATE_MOVING, // 运动 } xmedia_svp_motion_state; typedef struct { xmedia_svp_alg_type alg_type; // 检测类别 xmedia_svp_class_type class_type; // 检测结果类别 xmedia_float detect_score; // 检测模型结果置信度 xmedia_float classfier_score; // 分类器结果置信度 xmedia_s32 tracker_id; // 检测结果追踪id xmedia_u32 tracker_age; // 检测结果追踪age xmedia_svp_rect rect; // 检测结果目标框 xmedia_bool special_target; // 特殊目标 xmedia_float distance; // 检测目标距离 xmedia_svp_motion_state motion_state; // 运动状态 } xmedia_svp_detect_result; typedef struct { xmedia_float x; xmedia_float y; } xmedia_svp_point; typedef struct { xmedia_bool enable; // 框面积大小限制是否使能 xmedia_float min_area; // 最小框面积 xmedia_float max_area; // 最大框面积 } xmedia_svp_area_limit_attr; typedef struct { xmedia_bool enable; // 边缘过滤是否使能 xmedia_float w_edge; // 宽度过滤距离(框中心点至左右两边距离) xmedia_float h_edge; // 高度过滤距离(框中心点至上下两边距离) } xmedia_svp_edge_limit_attr; typedef enum { XMEDIA_SVP_TASK_UAV, XMEDIA_SVP_TASK_DETECT, XMEDIA_SVP_TASK_AOV, XMEDIA_SVP_TASK_DETECT_AND_KEYPOINT, XMEDIA_SVP_TASK_ADAS, XMEDIA_SVP_TASK_FIRE, XMEDIA_SVP_TASK_GESTURE, XMEDIA_SVP_TASK_FACE_RECOGNITON, XMEDIA_SVP_TASK_EMOTION_CLASSIFITION, XMEDIA_SVP_TASK_RCNN, XMEDIA_SVP_TASK_RCNN_720P, XMEDIA_SVP_TASK_COMPANION_ROBOT, XMEDIA_SVP_TASK_MULTI_DETECT, XMEDIA_SVP_TASK_DMS, XMEDIA_SVP_TASK_PLATE, XMEDIA_SVP_TASK_VEHICLE, XMEDIA_SVP_TASK_PLATE_VEHICLE, XMEDIA_SVP_TASK_PCNMV_FIRE, XMEDIA_SVP_TASK_MAX, } xmedia_svp_task_type; typedef struct { xmedia_video_frame_info *frame; xmedia_u8 frame_num; } xmedia_svp_task_input; typedef struct { xmedia_svp_model_type load_mode; // 模型加载模式 xmedia_svp_inputdata_format format; // 模型输入图像格式 xmedia_char *pathname; // 模型文件 xmedia_u8 *buf; // 模型存放的内存 xmedia_u32 len; // 模型的长度 xmedia_svp_alg_type alg_type; // 模型检测类别 xmedia_void *priv; // 模型私有数据 } xmedia_svp_modules; typedef struct { xmedia_svp_task_type task_type; // 任务类型 xmedia_svp_modules *modules; // 模型数据 xmedia_u8 module_num; // 模型数量 xmedia_void *priv; // 私有数据 } xmedia_svp_task_cfg; typedef struct { xmedia_float detect_threshold; // 置信度阈值，建议值0.65f xmedia_float classifier_threshold; // 分类器阈值，建议值0.8f xmedia_float iou_threshold; // iou相交比阈值，建议值0.5f xmedia_u32 max_target_num; // 最大目标数，最大值10 } xmedia_svp_detect_base_attr; typedef struct { xmedia_bool enable; // 追踪使能 xmedia_float max_feature_distance; // 追踪特征匹配距离最大阈值 xmedia_float min_feature_distance; // 追踪特征匹配距离最小阈值 xmedia_float max_iou_distance; // 追踪iou匹配距离阈值 xmedia_u32 iou2feature; // 追踪iou匹配变成确认态 xmedia_u32 max_age; // 追踪age阈值 } xmedia_svp_deepsort_attr; typedef struct { xmedia_svp_detect_base_attr person_attr; xmedia_svp_detect_base_attr head_attr; xmedia_svp_detect_base_attr gesture_attr; xmedia_svp_deepsort_attr deepsort_attr; xmedia_u32 focal_length_px; xmedia_u32 head_width_cm; xmedia_u32 palm_width_cm; } xmedia_svp_uav_attr; typedef struct { xmedia_svp_detect_result track_result; xmedia_svp_detect_result head_result; xmedia_svp_detect_result gesture_result; xmedia_float head_distance; xmedia_bool is_hover; xmedia_float gesture_distance; } xmedia_svp_uav_output; typedef struct { xmedia_float detect_threshold; // 置信度阈值，建议值0.65f xmedia_float classifier_threshold; // 分类器阈值，建议值0.8f xmedia_float iou_threshold; // iou相交比阈值，建议值0.5f xmedia_u32 max_target_num; // 最大目标数，最大值10 xmedia_bool bytetrack_enable; // 追踪开关 xmedia_bool motionless_filter_enable; // 运动状态检测开关 xmedia_float stillness_thres; // 由运动转为静止状态的灵敏度阈值，建议值0.9f xmedia_u8 movement_fps_thres; // 由运动转为静止状态的连续帧阈值，建议值5 xmedia_bool smart_venc_enable; xmedia_bool smart_ae_enable; xmedia_bool smart_venc_array[XMEDIA_SVP_MAX_VENC_CHN_NUM]; xmedia_bool smart_ae_array[XMEDIA_SVP_MAX_VI_PIPE_NUM]; } xmedia_svp_yolov5_attr; typedef struct { xmedia_u32 target_num; // 检测结果数量 xmedia_svp_detect_result targets[XMEDIA_SVP_MAX_TARGET_NUM]; // 检测结果 } xmedia_svp_yolov5_output; typedef struct { xmedia_float detect_threshold; // 置信度阈值，建议值0.65f xmedia_float classifier_threshold; // 分类器阈值，建议值0.8f xmedia_float iou_threshold; // iou相交比阈值，建议值0.5f xmedia_u32 max_target_num; // 最大目标数，最大值10 xmedia_bool bytetrack_enable; // 追踪开关 xmedia_bool motionless_filter_enable; // 运动状态检测开关 xmedia_bool aov_only_target; // 只获取目标有无 } xmedia_svp_aov_attr; typedef struct { xmedia_bool is_exist; xmedia_svp_yolov5_output detect_output; } xmedia_svp_aov_detect_output; typedef struct { xmedia_float x; xmedia_float y; xmedia_float score; } xmedia_svp_keypoint; typedef struct { xmedia_float detect_threshold; // 置信度阈值，建议值0.65f xmedia_float classifier_threshold; // 分类器阈值，建议值0.8f xmedia_float iou_threshold; // iou相交比阈值，建议值0.5f xmedia_u32 max_target_num; // 最大目标数，最大值10 xmedia_bool bytetrack_enable; // 追踪开关 xmedia_bool motionless_filter_enable; // 运动状态检测开关 xmedia_float stillness_thres; // 由运动转为静止状态的灵敏度阈值，建议值0.9f xmedia_u8 movement_fps_thres; // 由运动转为静止状态的连续帧阈值，建议值5 } xmedia_svp_yolov8_attr; typedef struct { xmedia_svp_alg_type alg_type; // 检测类别 xmedia_svp_class_type class_type; // 检测结果类别 xmedia_float detect_score; // 检测模型结果置信度 xmedia_float classfier_score; // 分类器结果置信度 xmedia_s32 tracker_id; // 检测结果追踪id xmedia_u32 tracker_age; // 检测结果追踪age xmedia_svp_rect rect; // 检测结果目标框 xmedia_svp_keypoint kpt[MAX_DETECT_KEYPOINT_NUM]; // 关键点坐标 xmedia_svp_motion_state motion_state; // 运动状态 } xmedia_svp_yolov8_result; typedef struct { xmedia_u32 target_num; // 检测结果数量 xmedia_svp_yolov8_result targets[XMEDIA_SVP_MAX_TARGET_NUM]; // 检测结果 } xmedia_svp_yolov8_output; typedef struct { xmedia_u32 focal_length_px; xmedia_u32 car_limit_angle; xmedia_float camera_height; xmedia_u32 plate_width; } xmedia_svp_adas_distance_attr; typedef struct { xmedia_svp_yolov5_attr car_attr; xmedia_svp_yolov5_attr plate_attr; xmedia_svp_yolov5_attr person_attr; xmedia_svp_yolov5_attr nmv_attr; xmedia_svp_adas_distance_attr distance_attr; } xmedia_svp_adas_attr; typedef struct { xmedia_s32 lane_point_num[XMEDIA_SVP_LANE_MAX_NUM]; xmedia_s32 lane_fitted_point_num[XMEDIA_SVP_LANE_MAX_NUM]; xmedia_svp_keypoint row_result[XMEDIA_SVP_LANE_AXIS_NUM][XMEDIA_SVP_LANE_MAX_ROW_POINT]; xmedia_svp_keypoint col_result[XMEDIA_SVP_LANE_AXIS_NUM][XMEDIA_SVP_LANE_MAX_COL_POINT]; xmedia_svp_keypoint row_fitted_result[XMEDIA_SVP_LANE_AXIS_NUM][XMEDIA_SVP_LANE_MAX_ROW_POINT]; xmedia_svp_keypoint col_fitted_result[XMEDIA_SVP_LANE_AXIS_NUM][XMEDIA_SVP_LANE_MAX_COL_POINT]; xmedia_svp_yolov5_output car_result; xmedia_svp_yolov5_output plate_result; xmedia_svp_yolov5_output person_result; xmedia_svp_yolov5_output nmv_result; xmedia_float coeffs[XMEDIA_SVP_LANE_MAX_NUM][XMEDIA_SVP_LANE_FIT_ORDER]; } xmedia_svp_adas_result; typedef struct { xmedia_u8 smoke_apear_count; xmedia_u8 smoke_loss_count; xmedia_float smoke_change_ratio; xmedia_u8 fire_loss_count; xmedia_float fire_change_ratio; xmedia_float fire_ratio_sum; xmedia_svp_yolov8_attr detect_attr; } xmedia_svp_fire_attr; typedef struct { xmedia_svp_rect gesture_rect; xmedia_s32 gesture_tracker_id; xmedia_float select_score[XMEDIA_SVP_MAX_TARGET_NUM]; xmedia_svp_yolov8_output result; } xmedia_svp_gesture_result; typedef enum { XMEDIA_SVP_CLASS_EMOTION_SMILE, XMEDIA_SVP_CLASS_EMOTION_NORMAL, XMEDIA_SVP_CLASS_EMOTION_UNKNOWN, } xmedia_svp_class_emotion_type; typedef struct { xmedia_float clockwise_thres; xmedia_float anticlockwise_thres; xmedia_float left_thres; xmedia_float right_thres; xmedia_float upward_thres; xmedia_float downward_thres; } xmedia_svp_face_pose_attr; typedef struct { xmedia_svp_yolov5_attr detect_attr; xmedia_u64 luma_score; xmedia_s32 ive_sobel_score; xmedia_svp_face_pose_attr fr_pose_attr; xmedia_float keypoint_score; } xmedia_svp_fr_attr; typedef struct { xmedia_float vector[XMEDIA_SVP_FEATURE_COL]; xmedia_svp_rect rect; xmedia_s32 fr_tracker_id; xmedia_float score; xmedia_svp_point keypoint_arry[XMEDIA_SVP_FR_KEYPOINT_NUM]; } xmedia_svp_fr_result; typedef struct { xmedia_svp_rect rect; xmedia_s32 face_tracker_id; xmedia_float score; xmedia_svp_point keypoint_arry[XMEDIA_SVP_FR_KEYPOINT_NUM]; xmedia_svp_class_emotion_type emotion_class; } xmedia_svp_face_attribute_result; typedef struct { xmedia_s32 face_num; xmedia_svp_fr_result fr_result[XMEDIA_SVP_MAX_TARGET_NUM]; } xmedia_svp_fr_output; typedef struct { xmedia_s32 face_num; xmedia_svp_face_attribute_result attribute_result[XMEDIA_SVP_MAX_TARGET_NUM]; } xmedia_svp_face_attribute_output; typedef struct { xmedia_u32 pet_num; xmedia_u32 kpt_num; xmedia_svp_detect_result pet_targets[XMEDIA_SVP_PETBOT_MAX_TARGET]; xmedia_svp_yolov8_result kpt_targets[XMEDIA_SVP_PETBOT_MAX_TARGET]; } xmedia_svp_petbot_output; typedef enum { XMEDIA_SVP_MEM_TYPE_BLOCK, XMEDIA_SVP_MEM_TYPE_AINR_SHARE, XMEDIA_SVP_MEM_TYPE_COMPLETE, XMEDIA_SVP_MEM_TYPE_MAX, } xmedia_svp_mem_reuse_type; typedef struct { xmedia_u32 size; xmedia_u64 phyaddr; xmedia_void *viraddr; } xmedia_svp_mem_info; typedef struct { xmedia_svp_mem_reuse_type reuse_type; xmedia_svp_mem_info workbuf_reuse_mem; xmedia_svp_mem_info input_reuse_mem; xmedia_svp_mem_info output_reuse_mem; } xmedia_svp_cfg; typedef struct { xmedia_float detect_threshold; // 置信度阈值，建议值0.65f xmedia_float classifier_threshold; // 分类器阈值，建议值0.8f xmedia_float iou_threshold; // iou相交比阈值，建议值0.5f xmedia_u8 max_target_num; // 单个检测任务的最大目标数，所有检测任务总和最大值50 } xmedia_svp_multi_base_attr; typedef struct { xmedia_svp_multi_base_attr car_attr; xmedia_svp_multi_base_attr person_attr; xmedia_svp_multi_base_attr nmv_attr; xmedia_svp_multi_base_attr face_attr; xmedia_svp_multi_base_attr pet_attr; xmedia_svp_multi_base_attr pkg_attr; xmedia_bool bytetrack_enable; // 追踪开关 xmedia_bool motionless_filter_enable; // 运动状态检测开关 xmedia_float stillness_thres; // 由运动转为静止状态的灵敏度阈值，建议值0.9f xmedia_u8 movement_fps_thres; // 由运动转为静止状态的连续帧阈值，建议值5 } xmedia_svp_multi_detect_attr; typedef struct { xmedia_svp_yolov8_attr detect_attr; xmedia_u8 warn_frame; // 警告帧数 xmedia_float iou_thres; // 相交阈值，建议值0.1f } xmedia_svp_dms_attr; typedef struct { xmedia_bool flag; // 状态标志位 xmedia_float detect_score; // 检测模型结果置信度 xmedia_svp_rect rect; // 检测结果目标框 } xmedia_svp_dms_state; typedef struct { xmedia_bool fatigue_flag; // 疲劳状态标志位 xmedia_u8 phone_num; // 手机检测数量，最大3 xmedia_svp_dms_state phone[XMEDIA_SVP_DMS_SPECIAL_MAX_NUM]; // 手机状态 xmedia_u8 cigar_num; // 香烟检测数量，最大3 xmedia_svp_dms_state cigar[XMEDIA_SVP_DMS_SPECIAL_MAX_NUM]; // 香烟状态 } xmedia_svp_dms_report; typedef struct { xmedia_svp_alg_type alg_type; // 检测类别 xmedia_svp_class_type class_type; // 检测结果类别 xmedia_float detect_score; // 检测模型结果置信度 xmedia_s32 tracker_id; // 检测结果追踪id xmedia_u32 tracker_age; // 检测结果追踪age xmedia_svp_rect rect; // 检测结果目标框 xmedia_bool special_target; // 特殊目标 xmedia_svp_keypoint kpt[MAX_DETECT_KEYPOINT_NUM]; // 关键点坐标 xmedia_svp_dms_report dms_report; } xmedia_svp_dms_result; typedef struct { xmedia_u32 target_num; // 检测结果数量 xmedia_svp_dms_result targets[XMEDIA_SVP_DMS_MAX_TARGET_NUM]; // 最大目标数5 } xmedia_svp_dms_output; typedef struct { xmedia_s32 x; // X坐标，采用万分比坐标 xmedia_s32 y; // Y坐标，采用万分比坐标 } xmedia_svp_vec2i; typedef struct { xmedia_s32 points_count; // 坐标点数量 xmedia_svp_vec2i points_array[XMEDIA_SVP_MAX_EDGE]; // 坐标点数组 } xmedia_svp_landmarks; typedef struct { xmedia_s32 zone_num; // 区域个数 xmedia_svp_landmarks zones[XMEDIA_SVP_MAX_REGION]; // 区域数组 } xmedia_svp_zone; typedef struct { xmedia_svp_yolov5_attr plate_detect_attr; xmedia_u32 plate_min_width; xmedia_u32 report_count; xmedia_float report_thres; xmedia_svp_zone detect_zone; } xmedia_svp_plate_reco_attr; // 车牌车辆方向 typedef enum { XMEDIA_SVP_DIRECTION_FORWARD, // 向前 XMEDIA_SVP_DIRECTION_BACKWARD, // 向后 XMEDIA_SVP_DIRECTION_UNKOWN = 99, // 未知 } xmedia_svp_direction; // 车牌颜色 typedef enum { XMEDIA_SVP_PLATE_COLOR_BLACK = 0, // 黑牌 XMEDIA_SVP_PLATE_COLOR_BULE = 1, // 蓝牌 XMEDIA_SVP_PLATE_COLOR_GREEN = 2, // 绿牌 XMEDIA_SVP_PLATE_COLOR_RESEVER = 3, // 保留 XMEDIA_SVP_PLATE_COLOR_WHITE = 4, // 白牌 XMEDIA_SVP_PLATE_COLOR_PURPLE = 5, // 紫牌 XMEDIA_SVP_PLATE_COLOR_YELLOW = 6, // 黄牌 XMEDIA_SVP_PLATE_COLOR_UNKNOWN = 7, // 未知 } xmedia_svp_plate_color; typedef struct { xmedia_s32 tracker_id; xmedia_char plate_char[XMEDIA_SVP_PLATE_CHAR_MAX_NUM]; xmedia_float char_score; xmedia_svp_plate_color color; // 车牌颜色 xmedia_float color_score; // 车牌颜色置信度 xmedia_svp_rect rect; // 车牌坐标 xmedia_svp_direction lpr_direction; // 车牌触发方向 } xmedia_svp_plate_result; typedef struct { xmedia_s32 target_num; xmedia_svp_plate_result targets[XMEDIA_SVP_PLATE_VEH_MAX_TARGET_NUM]; } xmedia_svp_plate_reco_output; typedef struct { xmedia_svp_yolov5_attr detect_attr; xmedia_u32 report_count; xmedia_float report_thres; xmedia_svp_zone detect_zone; } xmedia_svp_vehicle_reco_attr; // 车辆类型 typedef enum { XMEDIA_SVP_VEHICLE_TYPE_SEDAN, // 小车 XMEDIA_SVP_VEHICLE_TYPE_MINIVAN, // 小型货车 XMEDIA_SVP_VEHICLE_TYPE_TRUCK, // 卡车 XMEDIA_SVP_VEHICLE_TYPE_TANKER, // 罐车 XMEDIA_SVP_VEHICLE_TYPE_MINIBUS, // 巴士 XMEDIA_SVP_VEHICLE_TYPE_UNKNOWN = 99, // 未知 } xmedia_svp_vehicle_type; // 车辆颜色 typedef enum { XMEDIA_SVP_VEHICLE_COLOR_WHITE, // 白色 XMEDIA_SVP_VEHICLE_COLOR_RED, // 红色 XMEDIA_SVP_VEHICLE_COLOR_YELLOW, // 黄色 XMEDIA_SVP_VEHICLE_COLOR_BLACK, // 黑色 XMEDIA_SVP_VEHICLE_COLOR_BLUE, // 蓝色 XMEDIA_SVP_VEHICLE_COLOR_GREEN, // 绿色 XMEDIA_SVP_VEHICLE_COLOR_UNKNOWN = 99, // 未知 } xmedia_svp_vehicle_color; typedef struct { xmedia_s32 tracker_id; xmedia_svp_vehicle_color color; // 车辆颜色 xmedia_svp_vehicle_type type; // 车辆类型 xmedia_float color_scr; // 车辆颜色置信度 xmedia_float type_scr; // 车辆类型置信度 xmedia_svp_rect rect; // 车辆坐标 xmedia_svp_direction vehicle_direction; // 车辆触发方向 } xmedia_svp_vehicle_result; typedef struct { xmedia_s32 target_num; xmedia_svp_vehicle_result targets[XMEDIA_SVP_PLATE_VEH_MAX_TARGET_NUM]; } xmedia_svp_vehicle_reco_output; typedef struct { xmedia_svp_yolov5_attr plate_attr; xmedia_svp_yolov5_attr veh_attr; xmedia_u32 plate_min_width; xmedia_u32 report_count; xmedia_float report_thres; xmedia_svp_zone detect_zone; } xmedia_svp_plate_vehicle_attr; typedef struct { xmedia_s32 tracker_id; xmedia_svp_vehicle_color veh_color; // 车辆颜色 xmedia_svp_vehicle_type veh_type; // 车辆类型 xmedia_float veh_color_scr; // 车辆颜色置信度 xmedia_float veh_type_scr; // 车辆类型置信度 xmedia_svp_rect veh_rect; // 车辆坐标 xmedia_svp_direction veh_direction; // 车辆触发方向 xmedia_char plate_char[XMEDIA_SVP_PLATE_CHAR_MAX_NUM]; xmedia_float plate_score; xmedia_svp_plate_color plate_color; // 车牌颜色 xmedia_float plate_color_scr; // 车牌颜色置信度 xmedia_svp_rect plate_rect; // 车牌坐标 xmedia_svp_direction plate_direction; // 车牌触发方向 } xmedia_svp_plate_veh_result; typedef struct { xmedia_s32 target_num; xmedia_svp_plate_veh_result targets[XMEDIA_SVP_PLATE_VEH_MAX_TARGET_NUM]; } xmedia_svp_plate_vehicle_output; typedef struct { xmedia_svp_yolov8_attr detect_attr; xmedia_bool bytetrack_enable; // 追踪开关 xmedia_bool motionless_filter_enable; // 运动状态检测开关 xmedia_float stillness_thres; // 由运动转为静止状态的灵敏度阈值，建议值0.9f xmedia_u8 movement_fps_thres; // 由运动转为静止状态的连续帧阈值，建议值5 xmedia_u8 smoke_apear_count; xmedia_u8 smoke_loss_count; xmedia_float smoke_change_rate; xmedia_u8 fire_loss_count; xmedia_float fire_change_rate; xmedia_float fire_ratio_sum; } xmedia_svp_pcnmv_fire_attr; /* * 函数功能：初始化SVP * 函数参数： * 无 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_init(xmedia_void); /* * 函数功能：去初始化SVP * 函数参数： * 无 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_uninit(xmedia_void); /* * 函数功能：设置svp配置信息 * 函数参数： * svp_cfg - 输入参数，svp配置信息 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_set_config(xmedia_svp_cfg* svp_cfg); /* * 函数功能：获取svp配置信息 * 函数参数： * svp_cfg - 输入参数，svp配置信息 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_get_config(xmedia_svp_cfg* svp_cfg); /* * 函数功能：创建一路svp_task处理流程 * 函数参数： * handle - 输出参数，创建成功返回SVP句柄 * cfg - 输入参数，svp_task配置信息 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 * 注意： * 每个句柄对应一路svp_task处理流程 */ xmedia_s32 xmedia_svp_task_create(xmedia_s32 *handle, const xmedia_svp_task_cfg cfg); /* * 函数功能：设置svp_task处理流程的属性 * 函数参数： * handle - 输入参数，svp_task处理流程句柄 * task_attr - 输入参数，设置svp_task属性 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 * 注意： * 创建后，销毁前的任意时间均可调用 */ xmedia_s32 xmedia_svp_task_set_attr(xmedia_s32 handle, const xmedia_void *task_attr); /* * 函数功能：获取svp_task处理流程的属性 * 函数参数： * handle - 输入参数，svp_task处理流程句柄 * task_attr - 输出参数，获取的svp_task属性 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_task_get_attr(xmedia_s32 handle, xmedia_void *task_attr); /* * 函数功能：开始svp_task处理流程 * 函数参数： * handle - 输入参数，SVP处理流程句柄 * input - 输入参数，输入图像帧信息 * output - 输出参数，处理结果信息 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 * 注意： * 创建后，销毁前的任意时间均可调用 */ xmedia_s32 xmedia_svp_task_process(xmedia_s32 handle, const xmedia_svp_task_input *input, xmedia_void *output); /* * 函数功能：销毁一路svp_task处理流程 * 函数参数： * handle - 输入参数，SVP处理流程句柄 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_task_destroy(xmedia_s32 handle); /* * 函数功能：获取当前SVP版本信息 * 函数参数： * version - 输出参数，用于保存SVP版本信息 * len - 输入参数，version的长度 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_get_version(xmedia_char *version, const xmedia_s32 len); /* * 函数功能：重定向handle * 函数参数： * old_handle - 输入参数，旧的SVP处理流程句柄 * new_handle - 输出参数，重定向后得到的SVP处理流程句柄 * 返回值： * XMEDIA_SUCCESS - 函数执行成功 * XMEDIA_FAILURE - 函数执行失败 */ xmedia_s32 xmedia_svp_dup_handle(xmedia_s32 old_handle, xmedia_s32 *new_handle); #ifdef __cplusplus } #endif #endif