Skip to content

Latest commit

 

History

History
261 lines (221 loc) · 9.8 KB

disney_face_stylizing.md

File metadata and controls

261 lines (221 loc) · 9.8 KB

迪士尼人脸风格化

一、SDK功能

将图像或视频中检测到的人脸,生成对应的迪士尼风格人脸图像。支持同时处理多张人脸

API的输入、输出对象可参考VNN 数据结构
API的执行情况(是否成功、错误原因等)可参考 VNN_Result 对应的状态码表


二、技术规格

移动端

指标 参数
支持图片格式 BGRA、RGBA、RGB、NV12、NV21、YUV420
支持架构 armeabi-v7、arm64-v8a
Android系统版本 5.0+
iOS系统版本 9.0+
最大支持人脸数 5

PC端

指标 参数
支持图片格式 BGRA、RGBA、RGB、YUV420F
支持架构 x86(Win Only)、x86_64、arm64(MacOS Only)
Windows系统版本 Win 7+
MacOS系统版本 10.10+

三、资源依赖

3.1 头文件

vnn_face.h
vnn_faceparser.h
vnn_kit.h
vnn_define.h

3.2 模型文件

face_disney[1.0.0].vnnmodel

3.3 动态库

Android/Linux

libvnn_core.so
libvnn_kit.so
libvnn_face.so
libvnn_faceparser.so

iOS

Accelerate.framework
CoreVideo.framework
Foundation.framework
vnn_core_ios.framework
vnn_kit_ios.framework
vnn_face_ios.framework
vnn_faceparser_ios.framework

MacOS

Accelerate.framework
CoreVideo.framework
Cocoa.framework
vnn_core_osx.framework
vnn_kit_osx.framework
vnn_face_osx.framework
vnn_faceparser_osx.framework

Windows

vnn_core.dll
vnn_kit.dll
vnn_face.dll
vnn_faceparser.dll

四、相关说明

4.1 处理流程

pipline

4.2 Demo示例

Android: 链接
iOS: 链接
Windows: 链接
MaoOS: 链接
Linux: 链接


五、API文档

5.1 初始化 VNN_Create_CartFaceMaker

说明: 输入模型路径,完成SDK的初始化,获得用于调用后续功能的Handle。该函数是VNN_Create_FaceParser函数的别名

VNN_Result VNN_Create_CartFaceMaker( VNNHandle * handle, const int argc, const void * argv[] )
参数 含义
handle 函数调用成功后记录合法的索引,用于调用后续功能,类型为VNN_Handle*,调用成功后handle数值大于0,输出
argc 输入模型文件数,类型为const int,输入
argv 每个模型文件的具体路径,类型为const char*[ ],输入

返回值: VNN_Result,具体值参见 状态码表
调用示例:

VNN_Handle _handle;

std::string model = _modelpath + "/face_disney[1.0.0].vnnmodel";

const char* argv[] = {
	model.c_str(),
};

const int argc = sizeof(argv)/sizeof(argv[0]);

VNN_Result ret = VNN_Create_CartFaceMaker(&_handle, argc, argv);

5.2 迪士尼人脸风格化 VNN_Apply_CartFaceMaker_CPU

说明: 输入包含人脸的图像,输出迪士尼风格化后的一张或多张图像。该函数是VNN_Apply_FaceParser_CPU函数的别名

VNN_Result VNN_Apply_CartFaceMaker_CPU(VNNHandle handle, const void* in_image, const void* face_data, void* output)
参数 含义
handle SDK实例索引,类型为VNN_Handle,输入
in_image 输入图像,类型为 VNN_Image*,输入
face_data 每张人脸的关键点检测信息,类型为 VNN_FaceFrameDataArr*,输入
output 检测结果,类型为 VNN_ImageArr*,输出

返回值: VNN_Result,具体值参见 状态码表
调用示例:

#define FACE_CHANNEL (3)
#define FACE_HEIGHT (512)
#define FACE_WIDTH  (512)

// input: 视频帧或图像,类型 VNN_Image
// faceArr:人脸关键点检测结果,类型 VNN_FaceFrameDataArr

VNN_ImageArr faceImageArr;
faceImageArr.imgsNum = faceArr.facesNum;
if (!_faceBuffer) {
    // 按最大支持人脸数申请Buffer
    _faceBuffer = (unsigned char *) malloc(FACE_HEIGHT * FACE_WIDTH * VNN_FRAMEDATAARR_MAX_FACES_NUM);
}
// 按实际检测出的人脸数设置输出对象
for (int f = 0; f < faceImageArr.imgsNum ; f++) {
    faceImageArr.imgsArr[f].width = FACE_WIDTH; // 输出图像的宽
    faceImageArr.imgsArr[f].height = FACE_HEIGHT; // 输出图像的高
    faceImageArr.imgsArr[f].channels = FACE_CHANNEL; // 输出
    faceImageArr.imgsArr[f].pix_fmt = VNN_PIX_FMT_RGB888; // 格式HWC,RGB
    faceImageArr.imgsArr[f].data = _faceBuffer + f * FACE_CHANNEL * FACE_HEIGHT * FACE_WIDTH; // 对应Buffer的首地址
 }
 VNN_Apply_CartFaceMaker_CPU(_handle, &input, &faceArr, &faceImageArr);

5.3 资源释放 VNN_Destroy_CartFaceMaker

说明: 不再使用SDK,释放内存等资源。该函数是VNN_Destroy_FaceParser函数的别名

VNN_Result VNN_Destroy_CartFaceMaker(VNNHandle* handle)
参数 含义
handle SDK实例索引,成功释放资源后将被修改为0(无效值),类型为VNN_Handle*,输入&输出

返回值: VNN_Result,具体值参见 状态码表
调用示例:

VNN_Result ret = VNN_Destory_CartFaceMaker(&_handle);
// VNN_Result ret = VNN_Destory_Face(&_handle_face); // 如不再使用人脸关键点检测SDK,也一并释放资源
// VNN_Result ret = VNN_Destory_FaceParser(&_handle_mask); // 如不再使用人脸分割SDK,也一并释放资源

5.4 设置参数 VNN_Set_CartFaceMaker_Attr

说明: 设定SDK实例的运行参数。该函数是VNN_Set_FaceParser_Attr函数的别名

VNN_Result VNN_Set_CartFaceMaker_Attr( VNNHandle handle, const char * name, const void * value )
参数 含义
handle SDK实例索引,类型为VNN_Handle,输入
name 属性名,类型const char*,输入
value 属性值,类型参见下表,输入

合法属性名和属性值

属性名 属性含义 属性值 属性值类型
暂时为空

返回值: VNN_Result,具体值参见 状态码表
调用示例:

// 暂时为空

5.5 获取参数 VNN_Get_CartFaceMaker_Attr

说明: 获取SDK实例的运行参数。该函数是VNN_Get_FaceParser_Attr函数的别名

VNN_Result VNN_Get_CartFaceMaker_Attr( VNNHandle handle, const char * name, const void * value )
参数 含义
handle SDK实例索引,类型为VNN_Handle,输入
name 属性名,类型const char*,输入
value 属性值,类型参见下表,输出

合法属性名和属性值

属性名 属性含义 属性值 属性值类型
暂时为空

返回值: 类型为VNN_Result,具体值参见 状态码表
调用示例:

// 暂时为空

5.6 人脸关键点检测相关API

参考 链接

5.7 人脸分割相关API

参考 链接


六、更新记录

版本 日期 更新说明
v1.0.0 2021.12.07 初次发布