[发明专利]一种图像修复的方法、装置及电子设备

说明书

本发明提供了一种图像修复的方法、装置及电子设备，其中，所述方法包括：将待处理图像输入目标去噪网络，其中，所述目标去噪网络包括单帧网络和递归网络，所述待处理图像为待处理视频中的任一帧；通过所述单帧网络对所述待处理图像进行去压缩噪声处理，输出第一图像；根据先前帧图像的内容，通过所述递归网络对所述待处理图像进行去压缩噪声处理，输出第二图像，其中，所述先前帧图像为所述待处理图像在所述待处理视频中的前一帧图像；将所述第一图像和第二图像进行加权求和，输出针对所述待处理图像的去噪图像。用于去除视频压缩中的噪声，提高显示品质。

技术领域

本发明涉及图像处理技术领域，特别涉及一种图像修复的方法、装置及电子设备。

背景技术

为了避免视频占用较大的存储空间，以及提高传输速度，常常需要将视频进行压缩，然而，压缩过程中难免产生各种噪声，进而影响显示效果。

可见，如何去除视频压缩中的噪声，提高显示品质就变得极为重要。

发明内容

本发明提供了一种图像修复的方法、装置及电子设备，用于去除视频压缩中的噪声，提高显示品质。

第一方面，本发明实施例提供了一种图像修复的方法，包括：

将待处理图像输入目标去噪网络，其中，所述目标去噪网络包括单帧网络和递归网络，所述待处理图像为待处理视频中的任一帧；

通过所述单帧网络对所述待处理图像进行去压缩噪声处理，输出第一图像；

根据先前帧图像的内容，通过所述递归网络对所述待处理图像进行去压缩噪声处理，输出第二图像，其中，所述先前帧图像为所述待处理图像在所述待处理视频中的前一帧图像；

将所述第一图像和第二图像进行加权求和，输出针对所述待处理图像的去噪图像。

在一种可能的实现方式中，所述根据先前帧图像的内容，通过所述递归网络对所述待处理图像进行去压缩噪声处理，输出第二图像，包括：

通过所述递归网络中级联的第一卷积层、第一特征串联层、第一采样层对所述待处理图像进行去压缩噪声处理，输出第二图像。在一种可能的实现方式中，所述递归网络中所述第一卷积层包括第一子卷积层和第二子卷积层，所述第一特征串联层包括第一子特征串联层和第二子特征串联层，所述第一采样层包括第一下采样层和第一上采样层，所述通过所述递归网络中级联的第一卷积层、第一特征串联层、第一采样层对所述待处理图像进行去压缩噪声处理，输出第二图像，包括：

通过所述第一子特征串联层接收由所述单帧网络中各第二卷积层中第三子卷积层所提取的所述待处理图像的第一特征图，以及通过所述第一子特征串联层获取所述递归网络中与各所述第三子卷积层对应的所述第一子卷积层从所述先前帧图像中所提取的第二特征图；

通过所述第一子特征串联层对所述第一特征图和所述第二特征图进行串联操作，获得串联特征图；

通过所述第一子卷积层对所述串联特征图进行压缩，获得压缩后的特征图，所述压缩后的特征图为通过各所述第一子卷积层从所述待处理图像中所提取的所述第二特征图；

通过所述第一采样层中的第一下采样层，从所述压缩后的特征图中提取多个空间尺寸的特征图；

通过所述第一上采样层，确定与所述多个空间尺寸中相同空间尺寸的特征图；

通过所述第二子特征串联层对相同空间尺寸的特征图在特征维度上进行拼接，获得第一拼接特征图；

通过所述第二子卷积层对所述第一拼接特征图进行处理，输出所述第二图像。