[发明专利]一种图像恢复方法、装置、设备及可读存储介质

说明书

本发明公开了一种图像恢复方法、装置、设备及可读存储介质，涉及图像恢复领域，该方法包括：接收待恢复的图像，将图像输入头部神经网络映射得到特征值；将特征值和初始化隐变量输入隐层表示神经网络进行迭代得到精确特征值和迭代隐变量；其中，隐层表示神经网络包括特征神经网络和Anderson加速模块；判断特征神经网络是否满足压缩映射条件；若是，停止迭代，将对应精确特征值输入尾部神经网络映射得到恢复图像；若否，继续迭代，直至判断特征神经网络满足压缩映射条件。通过Anderson加速算法加快不动点映射的迭代，避免了需要对不动点映射进行大量迭代而导致图像恢复速度慢的问题。

技术领域

本发明涉及图像恢复领域，特别涉及一种图像恢复方法、装置、设备及计算机可读存储介质。

背景技术

图像恢复是指对低质量的输入图像经过恢复方法处理重建高质量输出图像，是计算机视觉中的基本问题之一。随着深度神经网络的快速发展和在人工智能领域多个领域取得突破性进展，基于深度不动点方法的隐层神经网络方法得到广泛关注和研究，在深度不动点方法中，不动点映射函数被深度神经网络参数化表示，通过不断地迭代该映射函数，即将深度不动点输出的特征表示再次作为输入特征，输入到深度不动点映射中，从而实现在不增加网络层数、参数的情况下，有效增加深度神经网络的等效深度，从而达到一个好的特征表示值，有利于提高恢复图像质量。

目前，虽然深度不动点方法在有限内存消耗下实现隐式无限深度的神经网络，但是在迭代逼近不动点过程中，需要对基于深度神经网络参数化的压缩映射函数进行大量的迭代，一般在百或者千次量级，导致推理速度慢。

发明内容

本发明的目的在于提供一种图像恢复方法、装置、设备及计算机可读存储介质，应用于图像恢复领域。本发明方法通过在隐层表示神经网络中引入Anderson加速，从而实现对深度不动点映射迭代次数的减少，提升深度不动点映射收敛的速度。

为实现上述目的，本发明提出一种图像恢复方法，包括：

接收待恢复的图像，将所述图像输入头部神经网络映射得到特征值；

将所述特征值和初始化隐变量输入隐层表示神经网络进行迭代得到精确特征值和迭代隐变量；其中，所述隐层表示神经网络包括特征神经网络和Anderson加速模块；

判断所述特征神经网络是否满足压缩映射条件；

若是，停止迭代，将对应精确特征值输入尾部神经网络映射得到恢复图像；

若否，继续迭代，直至判断所述特征神经网络满足所述压缩映射条件。

可选的，所述将所述特征值和初始化隐变量输入隐层表示神经网络进行迭代得到精确特征值和迭代隐变量，包括：

将所述特征值和所述初始化隐变量输入所述隐层表示神经网络进行迭代得到所述精确特征值和所述迭代隐变量；其中所述特征神经网络为残差结构卷积神经网络。

可选的，所述将所述特征值和初始化隐变量输入隐层表示神经网络进行迭代得到精确特征值和迭代隐变量，包括：

在所述隐层表示神经网络中，确定第一加权系数，确定第二加权系数，确定残差；

将所述特征值、所述初始化隐变量、所述第一加权系数、所述第二加权系数和所述残差输入所述隐层表示神经网络中的第一模型进行迭代得到所述精确特征值和所述迭代隐变量；其中，所述第一模型的数学表达式为：