6fdcb6a5
 Hu Chunming
初次提交,代码大体完成编写，完善中
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
#include "acl/acl.h"
#include "acl/ops/acl_dvpp.h"
#include "ai_engine_header.h"
#include <vector>
using namespace std;
struct HFResult{
float feature[HUMANREID_FEATURESIZE];
};
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};
};
|
