Blame view

src/PicAnalysis.h 1.62 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
  #include "village_inc.h"
52516fb9   Hu Chunming   添加授权代码;
15
  #include "atlas_licence.h"
e8beee4d   Hu Chunming   实现jni接口
16
  
d708ccb3   Hu Chunming   增加日志
17
18
  #include <mutex>
  
20396d5c   Hu Chunming   添加车头车尾算法
19
20
21
22
23
24
25
26
27
  using namespace std;
  
  
  class PicAnalysis
  {
  public:
      PicAnalysis(/* args */);
      ~PicAnalysis();
  
e8beee4d   Hu Chunming   实现jni接口
28
      int init(VillageParam param);
20396d5c   Hu Chunming   添加车头车尾算法
29
  
427201b2   Hu Chunming   完善结果返回
30
      vector<AnalysisResult> analysis_file(vector<string> file_path);
20396d5c   Hu Chunming   添加车头车尾算法
31
  
427201b2   Hu Chunming   完善结果返回
32
      vector<AnalysisResult> analysis_img(vector<sy_img> vec_img);
20396d5c   Hu Chunming   添加车头车尾算法
33
34
  
  private:
e8beee4d   Hu Chunming   实现jni接口
35
36
      int release();
  
52516fb9   Hu Chunming   添加授权代码;
37
38
39
      bool CheckTime();
      bool CheckLabel(int devId);
    	bool UpdateLabel();
427201b2   Hu Chunming   完善结果返回
40
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
41
  private:
20396d5c   Hu Chunming   添加车头车尾算法
42
43
44
45
46
47
      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;
15756629   Hu Chunming   添加clothes算法
48
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
49
50
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
f171c20a   Hu Chunming   添加moter_rainshed ...
51
52
      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;
2ae58093   Hu Chunming   添加road_seg算法
53
      RoadSegAnalysis m_road_seg_algorithm;
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
54
55
  
      CropUtil m_crop_util;
d708ccb3   Hu Chunming   增加日志
56
57
  
      std::mutex m_single_mtx;
52516fb9   Hu Chunming   添加授权代码;
58
59
60
61
  
      void *skt_handle {nullptr};//授权
  	atlas_licence_check_param check_param;//授权
  	int check_label {-1};//授权 -1=未授权  0=授权  1=授权check日已经check成功(每月一次check
20396d5c   Hu Chunming   添加车头车尾算法
62
  };