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