Blame view

src/village_inc.h 2.4 KB
e8beee4d   Hu Chunming   实现jni接口
1
2
3
4
  #ifndef __VILLAGE_INC_H__
  #define __VILLAGE_INC_H__
  
  #include <string>
427201b2   Hu Chunming   完善结果返回
5
6
  #include <vector>
  #include "vehicle_analysis.h"
e8beee4d   Hu Chunming   实现jni接口
7
  
581a68a4   Hu Chunming   修正parse_road无返回值导...
8
9
10
11
  #include "opencv2/opencv.hpp"
  #include "opencv2/imgcodecs/legacy/constants_c.h"
  #include "opencv2/imgproc/types_c.h"
  
e8beee4d   Hu Chunming   实现jni接口
12
13
  struct VillageParam {
      int dev_id;
581a68a4   Hu Chunming   修正parse_road无返回值导...
14
      std::string sdk_path;
e8beee4d   Hu Chunming   实现jni接口
15
16
  };
  
427201b2   Hu Chunming   完善结果返回
17
18
19
20
21
22
  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;               
f258e1c8   Hu Chunming   添加非机动车驾乘信息
23
  	int iColor;     // "棕", "橙", "灰", "白", "粉", "紫", "红", "绿", "蓝", "银", "青", "黄", "黑", ,"多色", "其他"
427201b2   Hu Chunming   完善结果返回
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
  } 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;	 
              
81e8a405   Hu Chunming   初步完成SDK
41
      int type;
5e69f8fc   Hu Chunming   补充字段;
42
  
6f9dffde   Hu Chunming   返回prob
43
      int vpt_type;       // person ;bike 自行车 ; motor 摩托车;tricycle 三轮车;car 汽车; bigbus  大巴;lorry 货车;tractor 拖拉机;midibus  面包/中巴
427201b2   Hu Chunming   完善结果返回
44
      int rainshed;		//是否安装雨棚, 0 有雨棚   1 无雨棚
6f9dffde   Hu Chunming   返回prob
45
      float rainshed_prob;
a1a053f2   Hu Chunming   修正人骑车逻辑错误
46
47
      int truck_manned{0};
      int motor_manned{0};
427201b2   Hu Chunming   完善结果返回
48
49
50
51
  
      int human_upper_color;
      int human_lower_color;
  
81e8a405   Hu Chunming   初步完成SDK
52
      int reverse_driving;    // 0 正常行驶   1 逆行   -1 未知
6f9dffde   Hu Chunming   返回prob
53
54
      int phoning;        // 骑车打电话 1 为是,其他为否
      float phoning_prob;
81e8a405   Hu Chunming   初步完成SDK
55
56
57
58
  
      int cross_line;         // 压实线, 1 黄实线 2 白实线  -1 其他
      int cross_diversion_line;   // 压导流线   1 压了  -1 未压
  
427201b2   Hu Chunming   完善结果返回
59
60
  } VehicleInfo;
  
06e0182f   Hu Chunming   修复src图片alignWidth...
61
62
  struct LineInfo{
      int line_type;
581a68a4   Hu Chunming   修正parse_road无返回值导...
63
      std::vector<cv::Point> vec_pt;
06e0182f   Hu Chunming   修复src图片alignWidth...
64
65
66
67
  };
  
  struct SegInfo {
      int seg_type;
581a68a4   Hu Chunming   修正parse_road无返回值导...
68
      std::vector<cv::Point> vec_pt;
06e0182f   Hu Chunming   修复src图片alignWidth...
69
70
  };
  
427201b2   Hu Chunming   完善结果返回
71
  typedef struct AnalysisResult {		                 
06e0182f   Hu Chunming   修复src图片alignWidth...
72
73
74
      std::vector<VehicleInfo> info;	             				              
      std::vector<LineInfo> vec_line;
      std::vector<SegInfo> vec_road;
427201b2   Hu Chunming   完善结果返回
75
  } AnalysisResult;   
e8beee4d   Hu Chunming   实现jni接口
76
77
  
  #endif          // __VILLAGE_INC_H__