Hu Chunming / mvpt_dq_ascend_arm

Blame view

src/ai_engine_module/HumanFeature.h 630 Bytes
Edit Raw Normal View History Permalink
d5345ee9   Hu Chunming   代码优化 
1
2
3
 
  #ifndef __AI_ENGINE_MODULE_HUMAN_FEATURE__
  #define __AI_ENGINE_MODULE_HUMAN_FEATURE__
6fdcb6a5   Hu Chunming   初次提交,代码大体完成编写，完善中 
4
5
6
7
8
9
10
11
12
13
 
  #include "acl/acl.h"
  #include "acl/ops/acl_dvpp.h"
  
  #include "ai_engine_header.h"
  
  #include <vector>
  
  using namespace std;
  
  struct HFResult{
d5345ee9   Hu Chunming   代码优化 
14
 
      float feature[HUMANREID_FEATURE_SIZE];
6fdcb6a5   Hu Chunming   初次提交,代码大体完成编写，完善中 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
  };
  
  class HumanFeature
  {
  public:
      HumanFeature(/* args */);
      ~HumanFeature();
  
      int init(int dev_id, string model_dir);
  
      vector<HFResult> process(vector<DeviceMemory*> vec_vptMem);
  
      void release();
  
  private:
      int m_devId; 
      aclrtContext m_algorthim_ctx;
      void* m_handle{nullptr};
d5345ee9   Hu Chunming   代码优化 
33
34
35
 
  };
  
  #endif      // __AI_ENGINE_MODULE_HUMAN_FEATURE__