Blame view

src/PicAnalysis.h 1.44 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
  
427201b2   Hu Chunming   完善结果返回
26
      vector<AnalysisResult> analysis_file(vector<string> file_path);
20396d5c   Hu Chunming   添加车头车尾算法
27
  
427201b2   Hu Chunming   完善结果返回
28
      vector<AnalysisResult> analysis_img(vector<sy_img> vec_img);
20396d5c   Hu Chunming   添加车头车尾算法
29
30
  
  private:
e8beee4d   Hu Chunming   实现jni接口
31
32
33
  
      int release();
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
34
35
36
37
      int human_analysis(vector<sy_img> vec_img);
  
      int check_motor_retrograde_motion(vector<sy_img> vec_img);
  
427201b2   Hu Chunming   完善结果返回
38
39
      vector<AnalysisResult> va_result2AnalysisResult(va_result* result, int batchsize);
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
40
  private:
20396d5c   Hu Chunming   添加车头车尾算法
41
42
43
44
45
46
      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;
15756629   Hu Chunming   添加clothes算法
47
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
48
49
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
f171c20a   Hu Chunming   添加moter_rainshed ...
50
51
      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;
2ae58093   Hu Chunming   添加road_seg算法
52
      RoadSegAnalysis m_road_seg_algorithm;
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
53
54
  
      CropUtil m_crop_util;
20396d5c   Hu Chunming   添加车头车尾算法
55
  };