[发明专利]基于空间自适应自监督学习的图像去噪方法

说明书

基于空间自适应自监督学习的图像去噪方法，属于图像复原技术领域。解决了现有自监督图像去噪方法存在对空间相关噪声去噪效果差的问题。本发明包括：步骤一、对噪声图像y进行区域特征提取，获得平坦区域监督图像和纹理区域监督图像步骤二、利用区域判别模块对平坦区域监督图像中每个像素点所在空间位置坐标(,j)处的平坦度自适应系数α(i,j)进行识别，从而确定的平坦度自适应系数矩阵α；步骤三、利用y、以及α对去噪模块进行训练，进行模型参数迭代更新，获得训练后的去噪模块；步骤四、利用训练后的去噪模块对待去噪图像进行去噪。本发明主要用于对相机采集的真实图像进行去噪。

技术领域

本发明涉及一种图像去噪方法，属于图像复原技术领域。

背景技术

图像去噪是底层视觉和图像处理领域的一个基本问题。随着深度学习的发展，深度神经网络成为主流的去噪方法。现有的图像去噪方法又分为有监督去噪方法和自监督去噪方法。

然而，有监督去噪方法中的深度神经网络在使用前需经过大量成对数据(噪声图-清晰图)训练，采集这样的数据对需要固定的拍摄环境和复杂的拍摄过程，费时费力。此外，不同的相机型号和光照条件下噪声分布也不同，每款相机的去噪模型都需要针对性地采集数据，采集数据的代价很高。

而自监督图像去噪方法在训练阶段仅使用噪声图像，由于待去噪的噪声图像本身即可用于训练，此类方法避免了训练数据采集的问题。但是，目前的自监督去噪方法适用的噪声类型仍局限于仿真噪声和空间无关噪声，其对仿真噪声和空间无关噪声两种类型的噪声去噪效果好，而对于空间相关噪声去除任务上去噪效果较差，因此，以上问题亟需解决。

发明内容

本发明目的是为了解决现有自监督图像去噪方法存在对空间相关噪声去噪效果差的问题，本发明提供了一种基于空间自适应自监督学习的图像去噪方法。

基于空间自适应自监督学习的图像去噪方法，该方法包括如下过程：

步骤一、对噪声图像y进行区域特征提取，获得平坦区域监督图像和纹理区域监督图像

步骤二、利用区域判别模块对平坦区域监督图像中每个像素点所在空间位置坐标(i,j)处的平坦度自适应系数α(i,j)进行识别，从而确定平坦区域监督图像的平坦度自适应系数矩阵α；

步骤三、利用噪声图像y、平坦区域监督图像纹理区域监督图像以及平坦度自适应系数矩阵α对去噪模块进行训练，使得去噪模块以自监督学习的方式进行参数更新，获得训练后的去噪模块；

步骤四、利用训练后的去噪模块对待去噪图像进行去噪。

优选的是，步骤一中，获得平坦区域监督图像的方式为：通过平坦区域监督提取模块对噪声图像y进行平坦区域进行特征提取，获得平坦区域监督图像

优选的是，平坦区域监督提取模块包括方向感受野模块、特征平移模块和特征融合模块；

方向感受野模块，用于提取噪声图像y的上、下、左、右四个方向的感受野，得到上、下、左、右四个方向感受野的特征图；

特征平移模块，用于对上、下、左、右四个方向感受野的特征图进行平移，得到平移后的特征图，且平移后的特征图中每个像素点所在位置处的感受野中不包括该像素点周围9×9区域内的像素；

特征融合模块，用于融合平移后的特征图，得到平坦区域的监督图像且平坦区域的监督图像中每个像素点所在位置处的感受野中在该像素点周围9×9区域内形成盲邻域。

优选的是，步骤一中，获得纹理区域监督图像的方式为：通过纹理区域监督提取模块对噪声图像y进行纹理区域特征提取，获得纹理区域监督图像