Blame view

src/PicAnalysis.h 1.41 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
  #include "./utils/CropUtil.h"
d708ccb3   Hu Chunming   增加日志
12
  #include "./utils/PathUtils.hpp"
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
13
  
e8beee4d   Hu Chunming   实现jni接口
14
15
  #include "village_inc.h"
  
d708ccb3   Hu Chunming   增加日志
16
17
  #include <mutex>
  
20396d5c   Hu Chunming   添加车头车尾算法
18
19
20
21
22
23
24
25
26
  using namespace std;
  
  
  class PicAnalysis
  {
  public:
      PicAnalysis(/* args */);
      ~PicAnalysis();
  
e8beee4d   Hu Chunming   实现jni接口
27
      int init(VillageParam param);
20396d5c   Hu Chunming   添加车头车尾算法
28
  
427201b2   Hu Chunming   完善结果返回
29
      vector<AnalysisResult> analysis_file(vector<string> file_path);
20396d5c   Hu Chunming   添加车头车尾算法
30
  
427201b2   Hu Chunming   完善结果返回
31
      vector<AnalysisResult> analysis_img(vector<sy_img> vec_img);
20396d5c   Hu Chunming   添加车头车尾算法
32
33
  
  private:
e8beee4d   Hu Chunming   实现jni接口
34
35
      int release();
  
427201b2   Hu Chunming   完善结果返回
36
37
      vector<AnalysisResult> va_result2AnalysisResult(va_result* result, int batchsize);
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
38
  private:
20396d5c   Hu Chunming   添加车头车尾算法
39
40
41
42
43
44
      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;
15756629   Hu Chunming   添加clothes算法
45
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
46
47
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
f171c20a   Hu Chunming   添加moter_rainshed ...
48
49
      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;
2ae58093   Hu Chunming   添加road_seg算法
50
      RoadSegAnalysis m_road_seg_algorithm;
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
51
52
  
      CropUtil m_crop_util;
d708ccb3   Hu Chunming   增加日志
53
54
  
      std::mutex m_single_mtx;
20396d5c   Hu Chunming   添加车头车尾算法
55
  };