HumanParsing.h
547 Bytes
#ifndef __AI_ENGINE_MODULE_HUMAN_PARSING__
#define __AI_ENGINE_MODULE_HUMAN_PARSING__
#include <vector>
#include "acl/acl.h"
#include "acl/ops/acl_dvpp.h"
#include "human_parsing.h"
#include "ai_engine_header.h"
using namespace std;
class HumanParsing
{
public:
HumanParsing(/* args */);
~HumanParsing();
int init(int dev_id, string model_dir);
vector<hp_analysis_res> process(vector<DeviceMemory*> );
void release();
private:
int m_devId;
aclrtContext m_algorthim_ctx;
void* m_handle{nullptr};
};
#endif