计算机与现代化 ›› 2023, Vol. 0 ›› Issue (03): 11-15.

• 图像处理 • 上一篇    下一篇

多尺度密集感受域的GAN图像去雾算法

  

  1. (1.曲阜师范大学网络空间安全学院,山东 曲阜 273165; 2.西南交通大学信息科学与技术学院,四川 成都 610000)
  • 出版日期:2023-04-17 发布日期:2023-04-17
  • 作者简介:尹相臣(2000—),男,山东临沂人,本科生,研究方向:图像处理,三维视觉,E-mail: 2889631270@qq.com; 陈思龙(2001—),男,四川达州人,本科生,研究方向:图像处理,E-mail: 313654034@qq.com; 李振凯(2000—),男,本科生,研究方向:软件工程,E-mail: 2737035853@qq.com; 张文进(2000—),男,山东昌乐人,本科生,研究方向:软件工程; 李桂青(1984—),女,山东烟台人,讲师,硕士,研究方向:计算机网络,图像处理。
  • 基金资助:
    国家级大学生创新训练项目(202110446262)

Algorithm of Multi-scale Dense Receptive Domain GAN lmage Dehazing

  1. (1. School of Cyber Science and Engineering, Qufu Normal University, Qufu 273165, China;
    2. School of Information Science and Technology, Southwest Jiaotong University, Chengdu 610000, China)
  • Online:2023-04-17 Published:2023-04-17

摘要: 基于先验的图像去雾算法依赖于大气散射模型,易受环境影响出现去雾不彻底、颜色失真等现象,针对上述问题本文基于深度学习,提出一种多尺度密集感受域的GAN图像去雾算法。首先构建一个多尺度学习的生成器网络,通过3种不同尺度提取图像的局部细节和全局信息后进行特征融合;然后通过感受密集块来增大感受野并获得丰富的上下文信息,将提取到的特征图在多个感受密集块中对特征进一步细化;接着使用一个多尺度的GAN判别器,由2个相同的子判别器D1和D2组成,2个子判别器联合指导生成器的训练;最后本文结合L1损失、感知损失和对抗损失,设计一个多元损失函数来收敛网络。在SOTS测试集上进行主观评价和客观评价,实验结果表明,本文算法取得了较优的效果,有效改善去雾不彻底的现象。

关键词: 图像去雾, GAN, 感受域, 多尺度学习

Abstract: The image dehazing based on prior relies on the atmospheric scattering model, which is susceptible to incomplete defogging and color distortion. Based on deep learning, this paper proposes a multi-scale dense receptive domain GAN image dehazing algorithm. Firstly, a multi-scale learning generator network is constructed to extract local details and global information of images through three different scales for feature fusion. Then, receptive dense blocks are used to increase receptive fields and obtain rich context information, and the extracted feature maps are further refined in multiple receptive dense blocks. Then,a multi-scale GAN discriminator is used, which consists of two identical sub-discriminators D1 and D2, and the two sub-discriminators jointly guide the generator training. Finally, L1 loss, perception loss and adversarial loss are combined to design a multivariate loss function to converge the network. The proposed algorithm is evaluated subjectively and objectively on SOTS test sets. The experimental results show that the proposed algorithm achieves better results and effectively, which improves the phenomenon of incomplete dehazing.

Key words: image dehazing, GAN, receptive field block, multi-scale learning