Blame view

src/ai_engine_module/HumanAnalysis.h 541 Bytes
4a273a4a   Hu Chunming   添加hcp和hp
1
2
3
  #include "include.h"
  #include "human_parsing.h"
  
531d44a1   Hu Chunming   添加日志
4
5
  
  // static std::string up_color[] = { "黑", "白", "灰", "红", "蓝", "黄", "绿", "多色", "其他" };
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
6
  struct BodyColorInfo{
b1547dfc   Hu Chunming   设置初始值,避免异常值影响
7
8
      int upper_body_color{14};
      int lower_body_color{14};
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
9
10
  };
  
4a273a4a   Hu Chunming   添加hcp和hp
11
12
13
14
15
16
  class HumanAnalysis
  {
  public:
      HumanAnalysis(/* args */);
      ~HumanAnalysis();
      
581a68a4   Hu Chunming   修正parse_road无返回值导...
17
      int init(int devId, std::string sdk_root);
4a273a4a   Hu Chunming   添加hcp和hp
18
  
b3012672   Hu Chunming   天啊及hp,road_seg子sd...
19
      vector<BodyColorInfo> detect(vector<sy_img> vec_img);
4a273a4a   Hu Chunming   添加hcp和hp
20
21
22
23
24
25
26
27
  
  private:
      int release();
  
  private:
      void* m_handle{nullptr};
      aclrtContext ctx{nullptr};
  };