[发明专利]掩膜图像生成方法、装置、电子设备和计算机可读介质

说明书

本公开的实施例公开了掩膜图像生成方法、装置、电子设备和计算机可读介质。该方法的一具体实施方式包括：提取破损图像中的边缘信息，得到破损图像的边缘图像；基于破损图像和边缘图像，对破损图像进行特征提取，得到破损图像的特征图像；将边缘图像进行变换，得到与特征图像尺寸匹配的缩小边缘图像；基于特征图像与缩小边缘图像，生成破损图像中破损区域的掩膜图像。该实施方式提供了一种生成破损图像中破损区域的掩膜图像的方法。该掩膜图像可以将破损区域定位出来，为后续修复破损图像中破损区域提供了基础。

技术领域

本公开的实施例涉及计算机技术领域，具体涉及掩膜图像生成方法、装置、电子设备和计算机可读介质。

背景技术

随着互联网的发展和以深度学习为核心的人工智能技术的普及，计算机视觉等技术已经应用到人们生活的各个领域。图像修复技术成是计算机视觉领域研究的一项重要的研究内容。图像修复技术通过对破损区域进行修复，得到修复后的图像。但是，图像修复技术无法直接从破损图像中确定需修复的破损图像中的破损区域，为了能够完成对破损图像的修复，存在着事先确定破损图像中破损区域的需要。

发明内容

本公开的内容部分用于以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。本公开的内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。

本公开的一些实施例提出了掩膜图像生成方法、装置、电子设备和计算机可读介质，来解决以上背景技术部分提到的技术问题。

第一方面，本公开的一些实施例提供了一种掩膜图像生成方法，该方法包括：提取破损图像中的边缘信息，得到破损图像的边缘图像；基于破损图像和边缘图像，对破损图像进行特征提取，得到破损图像的特征图像；将边缘图像进行变换，得到与特征图像尺寸匹配的缩小边缘图像；基于特征图像与缩小边缘图像，生成破损图像中破损区域的掩膜图像。

第二方面，本公开的一些实施例提供了一种掩膜图像生成装置，装置包括：边缘提取单元，被配置成提取破损图像中的边缘信息，得到破损图像的边缘图像；特征提取单元，被配置成基于破损图像和边缘图像，对破损图像进行特征提取，得到破损图像的特征图像；变换单元，被配置成将边缘图像进行变换，得到与特征图像尺寸匹配的缩小边缘图像；生成单元，被配置成基于特征图像与缩小边缘图像，生成破损图像中破损区域的掩膜图像。

第三方面，本公开的一些实施例提供了一种电子设备，包括：一个或多个处理器；存储装置，其上存储有一个或多个程序，当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现如第一方面中任一的方法。

第四方面，本公开的一些实施例提供了一种计算机可读介质，其上存储有计算机程序，其中，程序被处理器执行时实现如第一方面中任一的方法。

本公开的上述各个实施例具有如下有益效果：通过本公开的一些实施例的掩膜图像生成方法通过生成破损区域的掩膜图像，确定了破损图像中的破损区域。该掩膜图像可以确定到破损图像中的破损区域，为后续进行图像修复提供了基础。方便了后续的图像修复。具体来说，发明人发现，造成相关技术不能确定破损图像中破损区域的原因在于：相关技术并没有提供确定破损图像中破损区域的方法。基于此，本公开的一些实施例的掩膜图像生成方法通过提取破损图像中大量的边缘信息，得到像素之间的关联性。进而得到破损图像的主体信息，从而可以确定破损图像中的破损区域。进一步，生成破损图像中破损区域的掩膜图像。在此基础上，上述破损区域的掩膜图像确定了破损图像中的破损区域。因此，可以利用破损图像中的边缘信息，得到像素之间的关联性。利用像素之间的关联性生成破损图像中破损区域的掩膜图像。进而，确定破损图像中的破损区域。为后对破损图像中破损区域进行修复提供了基础。

附图说明