dface_attribute.h

Go to the documentation of this file.

// Hang Zhou Xiao Kong Cheng Xiang Ke Ji You Xian Gong Si is pleased to support DFACE SDK LIBRARY.
//
//  Copyright © 2017-2020, Hang Zhou Xiao Kong Cheng Xiang Ke Ji You Xian Gong Si Holding Limited//

#ifndef DFACE_ATTRIBUTE_H
#define DFACE_ATTRIBUTE_H

#include <stdio.h>
#include <vector>
#include <string>
#include "common.h"

namespace dface {

    class DFACE_EXPORTS DfaceAttribute {

    public:
        DfaceAttribute() {};

        virtual ~DfaceAttribute() { return; };


        virtual void predictAgeGender(unsigned char* data, int w, int h, int format, dface::Box &in_bbox, int &out_age, int &out_gender) = 0;


        virtual float predictMask(unsigned char* data, int w, int h, int format, dface::Box &in_bbox) = 0;
    };


    typedef DfaceAttribute* CreateAttribute(const std::string &model_path);

    typedef void DestroyAttribute(DfaceAttribute* dc);

}

#endif //DFACE_ATTRIBUTE_H