/************************************************************ * Version: vehicle_analysis_v0.0.0.190311.gpuonly * CopyRight: 中科视语（北京）科技有限公司 * UpdateDate: 20190311 * Content: drafting the first edition by ************************************************************/ #ifndef VEHICLEANALYSIS_H_ #define VEHICLEANALYSIS_H_ #ifdef _MSC_VER #ifdef VEHICLEANALYSIS_EXPORTS #define VEHICLEANALYSIS_API __declspec(dllexport) #else #define VEHICLEANALYSIS_API __declspec(dllimport) #endif #else #define VEHICLEANALYSIS_API __attribute__ ((visibility ("default"))) #endif #ifdef __cplusplus extern "C" { #endif #include "sy_common.h" //通用数据结构体定义 #include "vehicle_result.h" //车辆分析结果结构体定义 #ifndef __VEHICLE_INFO__ #define __VEHICLE_INFO__ typedef struct vehicle_info { //单车辆分析结果 vd_result vehicle_detect_res; //车(车头+车尾)检测结果 vd_result vehicle_win_detect_res; //车窗检测结果 vd_result vehicle_body_detect_res; //车身检测结果 vc_result vehicle_color_res; //车颜色 vr_result vehicle_recg_res; //车型识别 vplate_results vehicle_plate_det_recg_res; //车牌检测识别结果 v_pendant_d_result vehicle_pendant_det_res; //车属性 vid_result vehicle_illegal_det_res; //车违规 vf_result vehicle_fea_res; //以车搜车 //vs_result vehicle_special_res; //特殊品类车型（只有货车有输出结果） //mucktruckcover_result vehicle_mucktruckcover_res; //渣土车盖盖状态(车头：只有特殊品类车型中的渣土车和建筑垃圾车有输出结果，车尾：只有货车有输出结果) mta_result mta_res; //摩托车是否载人/摩托车驾驶人是否戴头盔/农用车是否载人 manned_result manned_res; //是否是货车三轮车载人 0-不是货车/不是三轮车/三轮车未载人/货车未载人，1-货车载人，2-三轮车载人 int type; //车的类别：0-车头，1-车尾，2-摩托车，3-三轮车 ，4-车身 //int obj_quality_idx;//聚档数据类型：0-无效数据，1-低质数据，2-优质数据 } vehicle_info; #endif #ifndef __VEHICLE_ANALYSIS_RESULT__ #define __VEHICLE_ANALYSIS_RESULT__ typedef struct va_result { //车辆分析结果 内存在外部申请 vehicle_info *info; //单车辆全部分析结果 int count; //车辆数量 } va_result; #endif #ifndef __DET_PARAM__ #define __DET_PARAM__ typedef struct det_param //检测算法参数 { int process_min_l; int process_max_l; float thresld; }det_param; #endif #ifndef __VA_PARAM__ #define __VA_PARAM__ typedef struct va_param { //算法配置参数 sy_command vehicle_detect_config; //是否启动车检测 sy_command vehicle_recg_config; //是否启动车型识别 sy_command vehicle_recg_supplement_config; //是否启动车型识别补充识别(车头车尾logo检测识别+车型15分类)（前提：启动车型识别vehicle_recg_config） sy_command vehicle_plate_det_recg_config; //是否启动车牌检测识别 sy_command vehicle_color_config; //是否启动车辆颜色识别 sy_command vehicle_pendant_det_config; //是否启动车辆属性识别 sy_command vehicle_illegal_config; //是否启动车违规行为检测 sy_command vehicle_feature_config; //是否启动车辆特征提取 //sy_command vehicle_special_config; //是否启动特殊品类车型识别。（前提：启动车型识别vehicle_recg_config） //sy_command vehicle_image_quality_config; //是否启动车辆图像清晰度识别 sy_command vehicle_motor_tricycle_analysis_config; //是否启动摩托车三轮车分析（摩托车是否载人/摩托车驾驶人是否戴头盔/农用车是否载人） //sy_command vehicle_motor_hs_output_config; //是否启动摩托车未戴头盔输出头肩坐标。（前提：启动摩托车三轮车分析vehicle_motor_tricycle_analysis_config） //sy_command vehicle_stain_vp_config; //是否启动遮挡车牌识别 //sy_command vehicle_muck_truck_cover_config; //是否启动渣土车盖盖识别。(前提：开启特殊品类车型识别vehicle_special_config) //sy_command vehicle_lorry_manned_config; //是否启动货车载人识别。（前提：开启车型识别vehicle_recg_config） sy_command vehicle_manned_config; //是否启动货车、三轮车载人识别。（前提：开启车型识别vehicle_recg_config） //算法初始化参数 int gpuid; //指定显卡id float vehicle_det_thresld=0.5; //车检测参数 int min_obj_size=200; //车头尾检测目标最小长边（建议取值范围100~200，长边小于该尺寸的过滤掉） float vehicle_plate_det_thresld=0.5; //车牌检测参数 float vehicle_attribute_det_thresld=0.5; //车属性检测参数 float vehicle_logo_det_thresld=0.1; //车logo检测参数 float vc_thresld=0.5; //车颜色阈值 char* dbPath; //车型识别车型数据路路径 <当前仅支持英文路径> char* models_Path; //所有模型的路径 <当前仅支持英文路径>//20210615 }va_param; #endif /************************************************************************* * function: va_init * purpose: init resources * param: [in] handle - handle [in] param - init param * return: success(0) or error code(<0) * notes: null *************************************************************************/ VEHICLEANALYSIS_API int va_init(void ** handle, va_param param); VEHICLEANALYSIS_API int va_acl_init();//仅调用一次，先调用va_acl_init，再调用va_init /************************************************************************* * function: va_release * purpose: release sources * param: [in] handle - handle * return: null * notes: null *************************************************************************/ VEHICLEANALYSIS_API void va_release(void ** handle); VEHICLEANALYSIS_API void va_acl_release();//仅调用一次，当所有的va_release都释放了，最后调用va_acl_release /************************************************************************* * function: va_get_version * purpose: get sdk version * param: null * return: null * notes: null *************************************************************************/ VEHICLEANALYSIS_API const char * va_get_version(); /************************************************************************* * function: va_batch * purpose: vehicle analysis batch * param: [in] handle - handle [in] img_data_array - data array [in] format - data format [in] batch_size - batch size [in] result - vehicle analysis result * return: success(0) or error code(<0) * notes: null *************************************************************************/ VEHICLEANALYSIS_API int va_batch(void * handle, sy_img * img_data_array, int batch_size, va_result *result); //img_data_array是device数据，batch_size==16时调用固定16模型，小于16时调用固定1模型。 VEHICLEANALYSIS_API int va_batch16_device(void * handle, sy_img * img_data_array, int batch_size, va_result *result); /************************************************************************* * FUNCTION: VA_ComputeSimilarity * PURPOSE: 比对相似度（通用） * PARAM: [in] Afea - 第一张图片特征 [in] Bfea - 第二张图片特征 [in] featuresize - 特征长度 * RETURN: 相似度 * NOTES: *************************************************************************/ VEHICLEANALYSIS_API float va_compute_similarity(float *Afea, float *Bfea, int featuresize); /************************************************************************* * FUNCTION: va_compute_similarity_byvafeature * PURPOSE: 比对相似度（针对va特征） * PARAM: [in] Afea - 第一张图片va特征 [in] Bfea - 第二张图片va特征 [in] featuresize - 特征长度VA_FEATURESIZE * RETURN: 相似度 * NOTES: *************************************************************************/ VEHICLEANALYSIS_API float va_compute_similarity_byvafeature(void * handle , float *Afea, float *Bfea, int featuresize); #ifdef __cplusplus }; #endif #endif