Blame view

src/PicAnalysis.h 858 Bytes
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"
20396d5c   Hu Chunming   添加车头车尾算法
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
  
  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算法
31
      VidClothes m_clothes_algorithm;
4a273a4a   Hu Chunming   添加hcp和hp
32
33
      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;
20396d5c   Hu Chunming   添加车头车尾算法
34
  };