基于深度学习的图像恢复技术研究与实现.docx

?

?

基于深度学习的图像恢复技术研究与实现

?

?

陆佳磊张永健

摘要：针对现有的基于传统算法的图像恢复技术无法有效地恢复出高质量图像问题，本文提出了基于深度学习技术的图像恢复方法。对现有的图像恢复网络进行改进，总体网路模型基于生成对抗网络，其中生成网络总体结构基于UNet网络，对上采样下采样的次数作调整，根据数据集的特点，减少了网络的下采样次数，防止因为网络的层数过多导致的特性信息的流失。特征提取模块中加入密集网络的特征提取模块，在减少网络参数的同时提升提取特征的能力。数据集使用了经过光加密系统后的CGH（ComputerGeneratedHolograms）计算生成全息图作为生成的噪声图像，并使用本文提出深度学习方法对该测试集进行恢复，对恢复的结果使用PSNR和SSIM指标进行评价，结果显示本文提出算法的有效性能。综上可知，本文算法所实现的模型能够有效地对现有的受损图像恢复并且恢复出高质量的图像，对图像恢复领域有着较高的实用价值和意义。

关键词：图像恢复;深度学习;图像去噪;计算生成全息图;图像光加密

：TP391??：A

：1009-3044（2021）14-0164-02

在如今信息技术高度发展的时代，信息安全变得尤为重要，光加密作为一种可靠的图像加密技术具有安全，高效等特点。在光加密系统中，随机相位掩膜面积有限，为一截断纯相位模板，因此傅立叶谱不可能是一纯相位函数，故生成的图像具有严重的图像噪声。直接对数据集使用光加密系统的噪声图像使用深度学习去噪需要大量的数据样本，无法达到实际使用的要求。

计算全息技术是光学技术与计算机技术结合的产物，通过计算机对于三维虚拟物体的光波和参考光的干涉进行模拟计算，通过波前调制设备加载干涉条纹重现三维虚拟物体。对数据集使用计算全息图后在使用图像加密系统对图像进行加密，计算全息图生成的图像特征类似，使用深度学习技术只需要学习相似的图像特征，这就能克服需要大量的数据样本来学习图像权重恢复图像的缺点，通过这种方法只需要少量的数据样本就能对解密后的噪声图像进行恢复。

基于传统的图像去噪算法对受损严重的图像无法有效的去噪。现有的深度学习算法如DnCNN和一些去雾去雨的网络针对的数据集噪声简单，而本文的噪声干扰较为严重，需要对特征提取模块和网络的总体结构重新设计来达到去除图像噪声的目的。

1相关工作

本文数据集通过仿真实验生成数据集。在生成数据的过程中，预先使用数据集转化为CGH图，再将CGH图像通过光加密系统加密解密，最后生成的解密图像伴随严重的噪声干扰，通过深度学习技术对解密后的噪声图去噪，恢复出高质量的无噪声图。

本文去噪技术的实现使用深度学习算法，其中网络基于GAN[1]（GenerativeAdversarialNetworks）生成对抗网络对图像进行去噪，与传统的深度学习模型不同的，生成对抗网络由生成网络和判别网络两个部分组合而成，生成网络的参数更新根据判别网络而来。判别网络和生成网络通过交替更新参数的方式训练，最后达到一个纳什平衡，生成最优的图像。生成网络拟合出接近于原图像的数据分布，可以生成质量较高的图像，在本文中，通过对抗训练的方式去除图像的噪声，最后拟合出高质量的无噪声图像。

生成网络基于UNet[2]拓扑结构，通过控制UNet网络的下采样参数来调整网络的最优结果。为了增加网络的性能，受到DenseBlock[3]密集网络模块启发，在网络的特征提取模块中加入了特征提取模块提升网络的性能。

根据EfficientNet[4]的启发，影响网络性能主要有三个指标，即网络的深度，网络的宽度和分辨率直接影响着网络的性能。网路的深度通过堆叠卷积层的数量实现，网络的宽度通过对不同尺度下的特征图提取特征实现，InceptionV3中通过对输入的特征图使用不同尺度的卷积核提取了特征的方式来增强网络的宽度。而网络的分辨率可以通过选择下采样的方式来控制，例如通过采样最大池化层和通过设置卷积核的步长来控制分辨率因素。基于该理论，本文通过深度参数来控制网络的深度，使用卷积核的方式会造成图像的失真问题，通过设置下采样的方式为最大池化层提取最有表征能力的特征。

2网络模型搭建

针对本文中CGH噪声图像特点，如何提取出图像中的有效特征信息是一个具有挑战性的问题，本文中的特征提取模块加入密集网络模块，每个密集网络模块使用由5个卷积层和归一化层、激活函数层基本构件BaseBlock组合而成。使用BN（Batchnormalization）层[5]可以加快网络的收敛速度，同时可以防止过拟合的过程，但是这也造成了网络中的参数过多。基于像素级别的图像恢复算法需要大量的網络参数来提升网络的性能，这样造成的网络的训练速度过慢，在模型中的加入1×1的卷积可以在不影响网络性能的情况下减少网