[发明专利]图像恢复方法、装置、电子设备和计算机存储介质

说明书

本发明实施例公开了一种图像恢复方法、装置、电子设备和计算机存储介质，其中，方法包括：对待恢复图像进行特征提取，获得待恢复图像的特征向量；利用策略神经网络，至少根据所述特征向量，获取对所述待恢复图像进行本次恢复的恢复策略向量；根据所述恢复策略向量在预定的多种恢复工具中确定本次恢复需要使用的恢复工具，利用确定的恢复工具对所述待恢复图像进行恢复；将本次恢复的图像作为新的待恢复图像，重复执行上述图像恢复操作，直至满足预定的停止条件。对于恢复失真的种类和顺序均未知的含有多种混合失真原因的失真图像，本发明方法可以得到显著的恢复效果。

技术领域

本发明涉及图像处理技术，尤其是一种图像恢复方法、装置、电子设备和计算机存储介质。

背景技术

获取图像的过程中，由于相机的曝光噪声、失焦模糊和图像压缩造成的失真等等原因可能导致图像质量下降，对质量下降的图像加以重建或恢复的处理为图像恢复处理，通过计算机处理。

发明内容

本发明实施例提供一种用于图像恢复的技术方案。

根据本发明实施例的一个方面，提供的一种图像恢复方法，包括：

对待恢复图像进行特征提取，获得所述待恢复图像的特征向量；

利用策略神经网络，至少根据所述特征向量，获取对所述待恢复图像进行本次恢复的恢复策略向量；

根据所述恢复策略向量在预定的多种恢复工具中确定本次恢复需要使用的恢复工具，利用确定的恢复工具对所述待恢复图像进行恢复；

将本次恢复的图像作为新的待恢复图像，重复执行上述图像恢复操作，直至满足预定的停止条件。

基于上述方法的另一实施例中，所述停止条件包括以下至少之一：

所述恢复策略向量对应的本次恢复策略为停止恢复；

已经执行的图像恢复操作达到预定的次数阈值。

基于上述方法的另一实施例中，其中，所述N+1维包括：分别表示预设的N种恢复工具中各恢复工具的使用概率的N维、以及表示停止恢复概率的1维，N为大于1的整数。

基于上述方法的另一实施例中，根据所述恢复策略向量在预定的多种恢复工具中确定本次恢复需要使用的恢复工具，包括：

根据所述恢复策略向量中各维度对应的恢复工具，确定最大使用概率值对应的恢复工具或者使用概率值大于设定概率阈值的至少一个维度对应的恢复工具，作为本次恢复需要使用的恢复工具。

基于上述方法的另一实施例中，所述至少根据所述特征向量，获取对所述待恢复图像进行本次恢复的恢复策略向量，包括：

根据所述特征向量与参考策略向量，获取对所述待恢复图像进行本次恢复的恢复策略向量；所述参考策略向量为对所述待恢复图像对应的原始图像或其恢复图像进行上一次图像恢复操作时使用的恢复策略向量；

所述以本次恢复的图像作为新的待恢复图像，重复执行上述图像恢复操作，包括：

将本次恢复的图像作为新的待恢复图像、将所述恢复策略向量作为新的参考策略向量，重复执行上述图像恢复操作。

基于上述方法的另一实施例中，根据所述特征向量与参考策略向量，获取对所述待恢复图像进行本次恢复的恢复策略向量，包括：

根据所述特征向量获取对所述待恢复图像进行本次恢复的候选恢复策略向量；

基于所述参考策略向量，由所述候选恢复策略向量获取对所述待恢复图像进行本次恢复的恢复策略向量。

基于上述方法的另一实施例中，基于所述参考策略向量，由所述候选恢复策略向量获取对所述待恢复图像进行本次恢复的恢复策略向量，包括：