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

说明书

本公开关于一种图像处理方法、装置、电子设备及存储介质，属于计算机技术领域。该方法包括：获取原始图像的待擦除区域，该待擦除区域包括待擦除对象；基于该待擦除区域，生成不包含该待擦除对象的擦除区域；对该擦除区域进行滤波，得到滤除了高频信息后的目标区域；将该原始图像中的该待擦除区域替换为该目标区域，得到目标图像。本公开通过将擦除该待擦除对象时遗留的残影、残留的噪声在滤波阶段被滤除，从而极大提升了对待擦除对象的擦除效果。

技术领域

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

背景技术

随着计算机技术的发展，用户能够在终端上对图像或视频中的图像帧进行二次编辑，达到二次创作生成内容的目的。通常，存在一类擦除图像中某个对象的处理需求，比如，擦除图像中的水印、遮挡视线的障碍物、误入镜头的路人、背景中影响美观的设施等。

以水印擦除情况为例，针对含水印图像，服务器部署基于Transformer框架的STTN(Spatial-Temporal Transformations Networks，时空图卷积神经网络)模型，来对含水印图像进行水印去除和内容填充，然而，STTN模型处理后的图像，在水印的擦除区域中会出现许多残影，即对图像中指定对象的擦除效果差。

发明内容

本公开提供一种图像处理方法、装置、电子设备及存储介质，以至少提升对图像中指定对象的擦除效果。本公开的技术方案如下：

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

获取原始图像的待擦除区域，所述待擦除区域包括待擦除对象；

基于所述待擦除区域，生成不包含所述待擦除对象的擦除区域；

对所述擦除区域进行滤波，得到滤除了高频信息后的目标区域；

将所述原始图像中的所述待擦除区域替换为所述目标区域，得到目标图像。

在一种可能实施方式中，所述对所述擦除区域进行滤波，得到滤除了高频信息后的目标区域包括：

将所述擦除区域输入保边滤波器，通过所述保边滤波器在保留所述擦除区域中边缘信息的情况下，滤除所述擦除区域中的高频信息，输出所述目标区域。

在一种可能实施方式中，所述保边滤波器为双边滤波器；

所述通过所述保边滤波器在保留所述擦除区域中边缘信息的情况下，滤除所述擦除区域中的高频信息，输出所述目标区域包括：

对所述擦除区域中的任一像素点，以所述像素点为中心，在所述擦除区域中采样得到所述像素点周围的多个邻域像素点；

确定每个邻域像素点的加权系数，所述加权系数基于所述邻域像素点和所述像素点的欧式距离和灰度差值确定得到；

基于每个邻域像素点的加权系数，对每个邻域像素点的像素值进行加权，将加权得到的各个像素值相加得到所述目标区域中与所述像素点位置相同的像素点的像素值。

在一种可能实施方式中，所述确定每个邻域像素点的加权系数包括：

基于所述邻域像素点与所述像素点之间的欧式距离，确定距离权重分量；

基于所述邻域像素点与所述像素点之间的灰度差值，确定色彩权重分量；

将所述距离权重分量和所述色彩权重分量相乘得到所述邻域像素点的加权系数。

在一种可能实施方式中，所述基于所述待擦除区域，生成不包含所述待擦除对象的擦除区域包括：

在所述待擦除区域中添加掩膜，所述掩膜用于覆盖所述待擦除对象；