Blame view

src/PicAnalysis.h 1.13 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
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
  
  using namespace std;
  
  
  class PicAnalysis
  {
  public:
      PicAnalysis(/* args */);
      ~PicAnalysis();
  
      int init(int dev_id);
  
      // todo 接口定义是二进制流的,这个地方有待修改
      int analysis_sync(vector<string> file_path);
  
      int release();
  
  private:
      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;
15756629   Hu Chunming   添加clothes算法
34
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
35
36
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
f171c20a   Hu Chunming   添加moter_rainshed ...
37
38
      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;
2ae58093   Hu Chunming   添加road_seg算法
39
      RoadSegAnalysis m_road_seg_algorithm;
20396d5c   Hu Chunming   添加车头车尾算法
40
  };