Hu Chunming / mvpt_dq_ascend_arm

#ifndef __AI_ENGINE_MODULE_HUMAN_FEATURE__
#define __AI_ENGINE_MODULE_HUMAN_FEATURE__

#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_FEATURE_SIZE];
};

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};
};

#endif      // __AI_ENGINE_MODULE_HUMAN_FEATURE__