[发明专利]一种基于图像噪声预测机制的图像超分辨率方法及系统

说明书

本发明公开了一种基于图像噪声预测机制的图像超分辨率方法及系统，方法包括：构建含有噪声干扰的训练数据集；构建基于噪声预测机制的图像超分辨率网络；将所述训练数据集中的图像输入到所述图像超分辨率网络中，完成基于噪声预测机制的图像超分辨率网络的训练；将待重建的低分辨率图像输入到基于噪声预测机制的图像超分辨率网络中，得到重建后的高分辨率重构图像。本发明有助于提高图像超分辨率效果，可广泛应用于计算机技术领域。

技术领域

本发明涉及计算机技术领域，尤其是一种基于图像噪声预测机制的图像超分辨率方法及系统。

背景技术

图像超分辨率技术是一种将低分辨率图像(Low Resolution，LR)重建为高分辨率图像(High Resolution，HR)的图像处理技术，在社会的各个行业存在着广泛的应用，如：卫星图像分析，医学图像分析，图像视频传输等。随着深度学习技术的发展，基于卷积神经网络(Convolution Neural Network，CNN)技术的图像超分辨率效果已经远远超过了基于插值的传统超分辨率方法。

基于深度学习技术的图像超分辨率方法可以大致分为两类。第一类是有参考的图像超分辨率方法，该类方法在输入LR图像的基础上，通过引入额外的HR图像作为参考，能够获取更丰富的图像特征，使得重建得到的高分辨率图像拥有更丰富的细节。然而，由于这类方法所需要同时提供的HR图像，一般在实际应用场景中难以获取，因此，该方法在实际应用中较少采用。

第二类则是无参考的图像超分辨率方法，该类方法只需要输入LR图像，就能够实现超分辨率图像的重建。然而，该类方法在训练时通常采用插值的方式在HR图像的基础上合成出一张LR的图像，然后再将该LR图像输入到卷积神经网络中进行训练。由于实际应用场景中的LR图像通常会受到噪声、压缩等干扰，使得实际应用中的LR图像与用于训练卷积神经网络的插值LR图像并不一致，导致在实际应用的场景中基于深度学习技术的图像超分辨率效果往往会有比较大的下降。

发明内容

有鉴于此，本发明实施例提供一种效果好的，基于图像噪声预测机制的图像超分辨率方法及系统。

本发明实施例的一方面提供了一种基于图像噪声预测机制的图像超分辨率方法，包括：

构建含有噪声干扰的训练数据集；

构建基于噪声预测机制的图像超分辨率网络；

将所述训练数据集中的图像输入到所述图像超分辨率网络中，完成基于噪声预测机制的图像超分辨率网络的训练；

将待重建的低分辨率图像输入到基于噪声预测机制的图像超分辨率网络中，得到重建后的高分辨率重构图像。

可选地，所述构建含有噪声干扰的训练数据集，包括：

将待处理的低分辨率图像转换为RGB格式的图像，得到的低质量RGB图像，将低质量RGB图像划分为若干个子图像；

针对每一个子图像，随机生成一个噪声加入到对应的子图像中；所述噪声的种类包括高斯噪声、均匀分布噪声或泊松噪声；

获取各个低质量RGB图像对应的高分辨率图像以及对应的噪声幅值，完成训练数据集的构建。

可选地，所述构建基于噪声预测机制的图像超分辨率网络中，所述图像超分辨率网络包括图像特征预提取模块、特征提取模块、噪声预测模块、上采样模块、图像特征精炼模块。

可选地，所述图像特征预提取模块的卷积核大小为3，输入通道为3，输出通道为64，padding填充为1，stride步幅为1。

可选地，所述噪声预测模块由第一卷积层、第一激活层、第二卷积层、第二激活层、第一全局池化层、第三激活层依次连接组成；