Blame view

src/ai_engine_module/HumanCarAnalysis.h 401 Bytes
4a273a4a   Hu Chunming   添加hcp和hp
1
2
3
  #include "include.h"
  #include "human_car_parsing.h"
  
81e8a405   Hu Chunming   初步完成SDK
4
5
6
7
8
  struct HumanCarResult {
      int type;
      float prob;
  };
  
4a273a4a   Hu Chunming   添加hcp和hp
9
10
11
12
13
14
15
16
  class HumanCarAnalysis
  {
  public:
      HumanCarAnalysis(/* args */);
      ~HumanCarAnalysis();
      
      int init(int devId);
  
81e8a405   Hu Chunming   初步完成SDK
17
      std::vector<HumanCarResult> detect(vector<sy_img> vec_img);
4a273a4a   Hu Chunming   添加hcp和hp
18
19
20
21
22
23
24
25
  
  private:
      int release();
  
  private:
      void* m_handle{nullptr};
      aclrtContext ctx{nullptr};
  };