计算机与现代化 ›› 2024, Vol. 0 ›› Issue (02): 56-63.doi: 10.3969/j.issn.1006-2475.2024.02.009

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

改进生成对抗网络的图像去雾算法

  

  1. (中北大学软件学院,山西 太原 030051)
  • 出版日期:2024-02-19 发布日期:2024-03-19
  • 作者简介:刘彦红(1998—),女,河北保定人,硕士研究生,研究方向:计算机视觉,图像处理,E-mail: 1754964517@qq.com; 杨秋翔(1969—),男,教授,硕士,研究方向:移动云计算与大数据技术及应用,E-mail: yangqx@nuc.edu.cn。

Image Dehazing Algorithm with Improved Generative Adversarial Network

  1. (College of Software, North University of China, Taiyuan 030051, China)
  • Online:2024-02-19 Published:2024-03-19

摘要: 摘要:雾霾天气下,可见光透过大气层时发生散射和吸收,导致图像质量变差、信息遮挡或丢失。基于此提出改进生成对抗网络(GAN)的图像去雾算法,该算法在生成器和鉴别器对抗中学习生成去雾图像。在生成器中,提出一种3行多列的多尺度融合注意力网络(Grid-G),引入通道注意力和像素注意力,分别从不同角度处理图像的厚雾区域和高频区域。在鉴别器中,引入图像中的高低频信息构建融合鉴别器(FD-F),将其作为额外先验判别图像的来源。在RESIDE数据集对合成数据和真实数据进行实验,实验结果表明本文算法在峰值信噪比和结构相似度等方面均优于其余对比算法,取得了更好的去雾效果,有效改善颜色失真等问题。

关键词: 关键词:图像去雾, 生成对抗网络, 融合鉴别器

Abstract: Abstract: In hazy weather, visible light scattering and absorption occur when passing through the atmosphere, resulting in poor image quality, information blocking or loss. Based on this, we propose an improved generative adversarial network (GAN) image dehazing algorithm, which learns to generate dehazed images in the generator and discriminator adversarial. In the generator, a three-row multi-column multi-scale fused attention network (Grid-G) is proposed to introduce channel attention and pixel attention to process the thick haze region and high frequency region of the image from different angles, respectively. In the discriminator, the high and low frequency information in the image is introduced to construct the fused discriminator (FD-F), which is used as a source of additional a priori discriminative images. Experiments on synthetic and real data in the RESIDE dataset show that the algorithm outperforms the rest of the comparison algorithms in terms of peak signal-to-noise ratio and structural similarity, achieves better dehazing effects, and effectively improves problems such as color distortion.

Key words: Key words: image dehazing, generative adversarial network, fusion discriminator

中图分类号: