matting_service.hpp

浏览该文件的文档.

 
#ifndef MATTING_SERVICE_HPP
#define MATTING_SERVICE_HPP
 
#ifdef WIN32
#define MATTING_SERVICE_API __declspec(dllexport)
#else // WIN32
#define MATTING_SERVICE_API
#endif // WIN32
 
#ifdef __cplusplus
extern "C" {
#endif // __cplusplus
 
typedef enum {
    MATTING_BGR,
    MATTING_RGB,
    MATTING_I420,
    MATTING_NV12,
    MATTING_NV21
} MattingColorFormat;
 
 
MATTING_SERVICE_API void MattingInit(const char* model_path);
 
MATTING_SERVICE_API void MattingPredict(unsigned char* img_ptr, unsigned char* mask_ptr, int height, int width, MattingColorFormat format, int degree, bool flip_x);
 
MATTING_SERVICE_API void MattingSetBackground(unsigned char* bg_ptr, int height, int width, MattingColorFormat format, int degree, bool flip_x);
 
MATTING_SERVICE_API void MattingActivateBlurMode(int alpha);
 
MATTING_SERVICE_API void MattingReset();
 
#ifdef ANDROID
MATTING_SERVICE_API void MattingCreate();
MATTING_SERVICE_API void MattingDestroy();
#endif // ANDROID
 
#ifdef __cplusplus
}
#endif // __cplusplus
 
#endif // MATTING_SERVICE_HPP