Blame view

src/PicAnalysis.h 1.4 KB
20396d5c   Hu Chunming   添加车头车尾算法
1
2
3
  #include "./ai_engine_module/include.h"
  #include "./ai_engine_module/VehicleAnalysis.h"
  #include "./ai_engine_module/VehicleHeadTail.h"
15756629   Hu Chunming   添加clothes算法
4
  #include "./ai_engine_module/VidClothes.h"
4a273a4a   Hu Chunming   添加hcp和hp
5
6
  #include "./ai_engine_module/HumanAnalysis.h"
  #include "./ai_engine_module/HumanCarAnalysis.h"
f171c20a   Hu Chunming   添加moter_rainshed ...
7
8
  #include "./ai_engine_module/MotorRainshedAnalysis.h"
  #include "./ai_engine_module/MotorPhoneAnalysis.h"
2ae58093   Hu Chunming   添加road_seg算法
9
  #include "./ai_engine_module/RoadSegAnalysis.h"
20396d5c   Hu Chunming   添加车头车尾算法
10
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
11
12
  #include "./utils/CropUtil.h"
  
e8beee4d   Hu Chunming   实现jni接口
13
14
  #include "village_inc.h"
  
20396d5c   Hu Chunming   添加车头车尾算法
15
16
17
18
19
20
21
22
23
  using namespace std;
  
  
  class PicAnalysis
  {
  public:
      PicAnalysis(/* args */);
      ~PicAnalysis();
  
e8beee4d   Hu Chunming   实现jni接口
24
      int init(VillageParam param);
20396d5c   Hu Chunming   添加车头车尾算法
25
26
  
      // todo 接口定义是二进制流的,这个地方有待修改
e8beee4d   Hu Chunming   实现jni接口
27
      va_result* analysis_file(vector<string> file_path);
20396d5c   Hu Chunming   添加车头车尾算法
28
  
e8beee4d   Hu Chunming   实现jni接口
29
      va_result* analysis_img(vector<sy_img> vec_img);
20396d5c   Hu Chunming   添加车头车尾算法
30
31
  
  private:
e8beee4d   Hu Chunming   实现jni接口
32
33
34
  
      int release();
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
35
36
37
38
39
      int human_analysis(vector<sy_img> vec_img);
  
      int check_motor_retrograde_motion(vector<sy_img> vec_img);
  
  private:
20396d5c   Hu Chunming   添加车头车尾算法
40
41
42
43
44
45
      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;
15756629   Hu Chunming   添加clothes算法
46
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
47
48
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
f171c20a   Hu Chunming   添加moter_rainshed ...
49
50
      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;
2ae58093   Hu Chunming   添加road_seg算法
51
      RoadSegAnalysis m_road_seg_algorithm;
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
52
53
  
      CropUtil m_crop_util;
20396d5c   Hu Chunming   添加车头车尾算法
54
  };