Hu Chunming / village_ascend_arm

Blame view

src/PicAnalysis.cpp 2.92 KB
Edit Raw Normal View History
20396d5c   Hu Chunming   添加车头车尾算法 
1
2
3
4
5
6
7
8
9
10
11
 
  #include "PicAnalysis.h"
  #include "./utils/logger.hpp"
  
  
  PicAnalysis::PicAnalysis(/* args */)
  {
      aclInit(nullptr);
  }
  
  PicAnalysis::~PicAnalysis()
  {
4a273a4a   Hu Chunming   添加hcp和hp 
12
 
      release();
20396d5c   Hu Chunming   添加车头车尾算法 
13
14
15
16
17
 
      aclFinalize();
  }
  
  int PicAnalysis::init(int dev_id) {
4a273a4a   Hu Chunming   添加hcp和hp 
18
19
20
 
      int ret = SY_FAILED;
  
      ret = m_vehicle_analysis.init(dev_id, 16);
20396d5c   Hu Chunming   添加车头车尾算法 
21
22
23
24
25
26
27
28
29
30
31
32
 
      if(0 != ret){
          return -1;
      }
  
      head_tail_param ht_param;
      ht_param.devId = dev_id;
      ht_param.max_batch = 16;
      ret = m_head_tail_algorithm.init(ht_param);
      if(0 != ret){
          return -1;
      }
15756629   Hu Chunming   添加clothes算法 
33
34
35
36
 
      ret = m_clothes_algorithm.init(dev_id);
      if(0 != ret){
          return -1;
      }
20396d5c   Hu Chunming   添加车头车尾算法 
37
4a273a4a   Hu Chunming   添加hcp和hp 
38
39
40
41
42
43
44
45
46
47
48
49
 
      ret = m_human_algorithm.init(dev_id);
      if(0 != ret){
          return -1;
      }
  
      ret = m_human_car_algorithm.init(dev_id);
      if(0 != ret){
          return -1;
      }
  
      ACL_CALL(aclrtCreateContext(&m_ctx, 0), ACL_SUCCESS, SY_FAILED);
  	ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
20396d5c   Hu Chunming   添加车头车尾算法 
50
51
52
53
54
55
56
57
58
 
  	ACL_CALL(aclrtCreateStream(&stream), ACL_SUCCESS, SY_FAILED);
      m_dvpp = new DvppProcess();
  	m_dvpp->InitResource(stream);
  
      return 0;
  }
  
  int PicAnalysis::analysis_sync(vector<string> vec_file_path){
4a273a4a   Hu Chunming   添加hcp和hp 
59
 
      ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
15756629   Hu Chunming   添加clothes算法 
60
20396d5c   Hu Chunming   添加车头车尾算法 
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
      const int batch_size = vec_file_path.size();
  
      vector<sy_img> vec_img;
  
      int ret = SY_FAILED;
  
      // ImageData 内部是智能指针，分析未处理完成前不得释放
      ImageData dvpp_data[batch_size];
      for (size_t i = 0; i < vec_file_path.size(); i++)
      {
          string file_path = vec_file_path[i];
          ImageData src;
          ret = Utils::ReadImageFile(src, file_path); //将二进制图像读入内存，并读取宽高信息
          if(ret != SY_SUCCESS){
              LOG_ERROR("ReadImageFile failed!");
              return -1;
          }
  
          ret = m_dvpp->CvtJpegToYuv420sp(dvpp_data[i], src); //解码
          if(ret != SY_SUCCESS){
              LOG_ERROR("CvtJpegToYuv420sp failed!");
              return -1;
          }
          
          sy_img img;
          img.w_ = dvpp_data[i].width;
          img.h_ = dvpp_data[i].height;
          img.data_ = dvpp_data[i].data.get();
  
          vec_img.push_back(img);
      }
15756629   Hu Chunming   添加clothes算法 
93
 
      m_vehicle_analysis.detect(vec_img);
20396d5c   Hu Chunming   添加车头车尾算法 
94
95
96
97
98
99
100
 
  
      vector<HeadTailResult> head_tail_result;
      ret = m_head_tail_algorithm.detect(vec_img, head_tail_result);
      if (0 != ret) {
          LOG_ERROR("m_head_tail_algorithm failed!");
          head_tail_result.clear();
      }
15756629   Hu Chunming   添加clothes算法 
101
102
 
  
      m_clothes_algorithm.detect(vec_img);
4a273a4a   Hu Chunming   添加hcp和hp 
103
104
105
106
 
  
      m_human_algorithm.detect(vec_img);
  
      m_human_car_algorithm.detect(vec_img);
20396d5c   Hu Chunming   添加车头车尾算法 
107
4a273a4a   Hu Chunming   添加hcp和hp 
108
109
 
      LOG_INFO("analysis_sync finished!");
20396d5c   Hu Chunming   添加车头车尾算法 
110
111
112
113
 
      return 0;
  }
  
  int PicAnalysis::release() {
4a273a4a   Hu Chunming   添加hcp和hp 
114
115
 
  
      ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
20396d5c   Hu Chunming   添加车头车尾算法 
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
 
  
      delete m_dvpp;
      m_dvpp = nullptr;
  
      if (stream != nullptr) {
          int ret = aclrtDestroyStream(stream);
          if (ret != ACL_SUCCESS) {
              LOG_ERROR("destroy stream failed");
          }
          stream = nullptr;
      }
  
      aclrtDestroyContext(m_ctx);
  
      return 0;
  }