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

说明书

本申请实施例提供了图像处理方法、装置、电子设备及存储介质，该方法包括：获取待处理的可见光图像及可见光图像对应的深度图像；根据待处理目标在可见光图像中的位置区域，确定替换图案的形状及目标替换区域；根据目标替换区域及深度图像，计算目标替换区域各点的法向量分量；根据目标替换区域各点的法向量分量，对替换图案的形状进行调整，得到目标图案；将目标图案覆盖到可见光图像的目标替换区域中。利用深度图像中的深度信息，计算需要替换的目标替换区域各点的法向量分量，并利用各点的法向量分量对替换图案的表面进行形变处理，考虑了三维场景中目标的形状，得到的目标图案更加接近真实的形状，替换后的图像效果更加真实。

技术领域

本申请涉及图像处理技术领域，特别是涉及图像处理方法、装置、电子设备及存储介质。

背景技术

随着计算机视觉技术的发展，特别是深度神经网络的出现，使得基于计算机的图像内指定目标的自动替换成为可能。

对图像内指定目标进行替换，一种典型的场景是对图像中的人体进行虚拟换装。现有技术中，利用深度神经网络识别出图像中人体的位置及形状，然后生成人体形状的服装，并覆盖到人体对应的位置，从而完成人体换装。

但是采用上述方法，只是模拟了图像二维平面内服装的形状，导致图像区域替换之后的图像效果差。

发明内容

本申请实施例的目的在于提供一种图像处理方法、装置、电子设备及存储介质，以实现提高图像区域替换之后的图像效果。具体技术方案如下：

第一方面，本申请实施例提供了一种图像处理方法，所述方法包括：

获取待处理的可见光图像及所述可见光图像对应的深度图像，其中，所述可见光图像中包括待处理目标；

根据所述待处理目标在所述可见光图像中的位置区域，确定替换图案的形状及目标替换区域；

根据所述目标替换区域及所述深度图像，计算所述目标替换区域各点的法向量分量，其中，每个所述点对应一个或多个像素点；

根据所述目标替换区域各点的法向量分量，对所述替换图案的形状进行调整，得到目标图案；

将所述目标图案覆盖到所述可见光图像的所述目标替换区域中。

在一种可能的实施方式中，所述根据所述待处理目标在所述可见光图像中的位置区域，确定替换图案的形状及目标替换区域，包括：

根据所述可见光图像的三分图和所述可见光图像，确定所述待处理目标在所述可见光图像中的位置区域，作为待处理目标区域，其中，所述三分图包括前景像素、背景像素以及不确定区域像素；

根据所述待处理目标区域中的关键点，确定目标替换区域及替换图案的形状。

在一种可能的实施方式中，所述根据所述可见光图像的三分图和所述可见光图像，确定所述待处理目标在所述可见光图像中的位置区域，作为待处理目标区域，包括：

对所述可见光图像进行前景像素及背景像素的分割，得到分割图像，其中，所述待处理目标对应所述前景像素；

对所述分割图像进行边缘检测，确定所述前景像素的边缘图像区域；

在所述分割图像中对所述边缘图像区域进行膨胀操作，得到不确定区域像素；

将包括所述前景像素、所述背景像素以及所述不确定区域像素的三分图及所述可见光图像输入到预先训练的深度神经网络中进行分析，得到所述待处理目标在所述可见光图像中的位置区域，作为待处理目标区域。

在一种可能的实施方式中，所述根据所述目标替换区域及所述深度图像，计算所述目标替换区域各点的法向量分量，包括：

根据所述深度图像，确定所述目标替换区域中各点的深度信息；