[发明专利]处理图像的方法、装置、电子设备、及存储介质

说明书

本公开实施例公开了一种处理图像的方法、装置、电子设备、及存储介质，方法包括：获取包括人脸的原始图像；获取所述原始图像中待处理的人脸图像；将所述人脸图像输入至预先训练的斑痘检测模型，得到所述人脸图像对应的斑痘概率矩阵；对所述斑痘概率矩阵进行反变换得到所述原始图像对应的第二斑痘概率矩阵；根据设定阈值对所述第二斑痘概率矩阵进行阈值化得到二值的阈值矩阵；根据所述阈值矩阵对所述原始图像进行修复处理。本公开实施例的技术方案能够在对人脸上的斑痘修复时避免照片失真。

技术领域

本公开实施例涉及计算机技术领域，具体涉及一种处理图像的方法、装置、电子设备、及存储介质。

背景技术

随着用户终端(例如智能手机)拍照功能日益强大，越来越多的用户喜欢使用用户终端来进行拍照，特别是有些爱美的用户喜欢使用手机的美颜相机拍照。

用户使用手机的美颜相机拍照时，需要即时对原始照片祛除面部斑点和痘印，为用户实时呈现无斑痘的靓照。而目前的美颜相机在祛除斑点和痘印后得到的美颜图像与真实图像相比往往失真较大，效果普遍不好。

发明内容

有鉴于此，本公开实施例提供一种处理图像的方法、装置、电子设备、及存储介质，以在修复人脸上的斑痘时避免照片失真。

本公开实施例的其他特性和优点将通过下面的详细描述变得显然，或部分地通过本公开实施例的实践而习得。

第一方面，本公开实施例提供了一种处理图像的方法，包括：

获取包括人脸的原始图像；

获取所述原始图像中待处理的人脸图像；

将所述人脸图像输入至预先训练的斑痘检测模型，得到所述人脸图像对应的斑痘概率矩阵，所述斑痘概率矩阵的尺寸与所述人脸图像的尺寸相同，所述斑痘概率矩阵的元素表示所述人脸图像中对应位置像素是斑痘的概率值；

对所述斑痘概率矩阵进行反变换得到所述原始图像对应的第二斑痘概率矩阵，所述第二斑痘概率矩阵的尺寸与所述原始图像的尺寸相同；

根据设定阈值对所述第二斑痘概率矩阵进行阈值化得到二值的阈值矩阵；

根据所述阈值矩阵对所述原始图像进行修复处理。

第二方面，本公开实施例还提供了一种处理图像的装置，包括：

原始图像获取单元，用于获取包括人脸的原始图像；

人脸图像获取单元，用于获取所述原始图像中待处理的人脸图像；

模型检测单元，用于将所述人脸图像输入至预先训练的斑痘检测模型，得到所述人脸图像对应的斑痘概率矩阵，所述斑痘概率矩阵的尺寸与所述人脸图像的尺寸相同，所述斑痘概率矩阵的元素表示所述人脸图像中对应位置像素是斑痘的概率值；

反变换单元，用于对所述斑痘概率矩阵进行反变换得到所述原始图像对应的第二斑痘概率矩阵，所述第二斑痘概率矩阵的尺寸与所述原始图像的尺寸相同；

二值化单元，用于根据设定阈值对所述第二斑痘概率矩阵进行阈值化得到二值的阈值矩阵；

图像修复单元，用于根据所述阈值矩阵对所述原始图像进行修复处理。

第三方面，本公开实施例还提供了一种电子设备，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如第一方面中任一项所述方法的指令。