village_inc.h 1.82 KB
#ifndef __VILLAGE_INC_H__
#define __VILLAGE_INC_H__

#include <string>
#include <vector>
#include "vehicle_analysis.h"

struct VillageParam {
    int dev_id;
    std::string db_path;
    std::string model_path;
};

typedef struct pendant_info
{
	sy_rect rect;
	int index;		                     //���� 0-driver 1-face 2-belt 3-sunshield 4-tag 5-decoration 6-napkinbox 7-zhuanjt 8-callPhone 9-sunRoof 10-holder 11-smoke
	float confidence;	 
    int driver_copilot_info;               
	int iColor;     //13类:"棕", "橙", "灰", "白", "粉", "紫", "红", "绿", "蓝", "银", "青", "黄", "黑"
} pendant_info;

typedef struct VehicleInfo {                   	 
    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;   	 
    std::vector<pendant_info> vehicle_pendant_det_res; 
    vid_result vehicle_illegal_det_res;          	
    vf_result vehicle_fea_res;                   	
    
    mta_result mta_res;                           

    manned_result manned_res;	 
            
    int type;                                    
    int vpt_type;
    int rainshed;		//是否安装雨棚, 0 有雨棚   1 无雨棚
    int manned;

    int human_upper_color;
    int human_lower_color;

} VehicleInfo;

struct LineInfo{
    int line_type;
    std::vector<sy_point> vec_pt;
};

struct SegInfo {
    int seg_type;
    std::vector<sy_point> vec_pt;
};

typedef struct AnalysisResult {		                 
    std::vector<VehicleInfo> info;	             				              
    std::vector<LineInfo> vec_line;
    std::vector<SegInfo> vec_road;
} AnalysisResult;   

#endif          // __VILLAGE_INC_H__