Hu Chunming / village_ascend_arm

#include "include.h"
#include "human_parsing.h"


// static std::string up_color[] = { "黑", "白", "灰", "红", "蓝", "黄", "绿", "多色", "其他" };
struct BodyColorInfo{
    int upper_body_color{14};
    int lower_body_color{14};
};

class HumanAnalysis
{
public:
    HumanAnalysis(/* args */);
    ~HumanAnalysis();
    
    int init(int devId, std::string sdk_root);

    vector<BodyColorInfo> detect(vector<sy_img> vec_img);

private:
    int release();

private:
    void* m_handle{nullptr};
    aclrtContext ctx{nullptr};
};