beautyPlugin/MakeBeautiful.py

"""

@author: cuny

@file: MakeBeautiful.py

@time: 2022/7/7 20:23

@description: 

美颜工具集合文件，作为暴露在外的插件接口

"""
from .GrindSkin import grindSkin
from .MakeWhiter import MakeWhiter
from .ThinFace import thinFace
import numpy as np


def makeBeautiful(input_image: np.ndarray,

                  landmark,

                  thinStrength: int,

                  thinPlace: int,

                  grindStrength: int,

                  whiterStrength: int

                  ) -> np.ndarray:
    """

    美颜工具的接口函数，用于实现美颜效果

    Args:

        input_image: 输入的图像

        landmark: 瘦脸需要的人脸关键点信息，为fd68返回的第二个参数

        thinStrength: 瘦脸强度，为0-10（如果更高其实也没什么问题），当强度为0或者更低时，则不瘦脸

        thinPlace: 选择瘦脸区域，为0-2之间的值，越大瘦脸的点越靠下

        grindStrength: 磨皮强度，为0-10（如果更高其实也没什么问题），当强度为0或者更低时，则不磨皮

        whiterStrength: 美白强度，为0-10（如果更高其实也没什么问题），当强度为0或者更低时，则不美白

    Returns:

        output_image 输出图像

    """
    try:
        _, _, _ = input_image.shape
    except ValueError:
        raise TypeError("输入图像必须为3通道或者4通道!")
    # 三通道或者四通道图像
    # 首先进行瘦脸
    input_image = thinFace(input_image, landmark, place=thinPlace, strength=thinStrength)
    # 其次进行磨皮
    input_image = grindSkin(src=input_image, strength=grindStrength)
    # 最后进行美白
    makeWhiter = MakeWhiter()
    input_image = makeWhiter.run(input_image, strength=whiterStrength)
    return input_image