付星堡, 蔡琼, 陈国清, 赖远哲, 陈玉. 改进CycleGAN的图像去雾网络[J]. 微电子学与计算机, 2021, 38(8): 87-94.
引用本文: 付星堡, 蔡琼, 陈国清, 赖远哲, 陈玉. 改进CycleGAN的图像去雾网络[J]. 微电子学与计算机, 2021, 38(8): 87-94.
FU Xingbao, CAI Qiong, CHEN Guoqing, LAI Yuanzhe, CHEN Yu. Image dehazing network based on improved CycleGAN[J]. Microelectronics & Computer, 2021, 38(8): 87-94.
Citation: FU Xingbao, CAI Qiong, CHEN Guoqing, LAI Yuanzhe, CHEN Yu. Image dehazing network based on improved CycleGAN[J]. Microelectronics & Computer, 2021, 38(8): 87-94.

改进CycleGAN的图像去雾网络

Image dehazing network based on improved CycleGAN

  • 摘要: 针对传统CycleGAN在图像去雾后出现模糊和颜色失真等问题,给出了一种改进CycleGAN的图像去雾网络.所提CycleGAN的生成器包括特征提取、特征融合和图像复原三个子网络.图像特征提取子网络用于提取图像的内容特征和风格特征,特征融合子网络利用两种不同的注意力机制分别对提取到的内容特征和风格特征进行融合,图像复原子网络将融合后的图像特征还原成无雾图像.与传统的CycleGAN和已有的去雾网络相比,所提网络对合成图像和真实图像均可取得理想的去雾结果,有效解决了传统CycleGAN在图像去雾后出现的模糊和颜色失真的问题.

     

    Abstract: In view of the blur and color distortion in traditional CycleGAN image dehazing, animproved CycleGANImage dehazing network is proposed. The generator of present CycleGAN consists three parts: feature extraction sub-network, feature fusion sub-network and image restoration sub-network. The image feature extraction sub network is used to extract the content features and style features of the image.The feature fusion sub network uses two different attention mechanisms to fuse the extracted content features and style features. The image complex atom network restores the fused image features to a haze free image.Compared with traditional CycleGAN and existing defogging networks, the proposed network can achieve ideal dehazing results for both synthetic images and real images, and effectively solves the problems of blurring and color distortion in traditional CycleGAN images after dehazing.

     

/

返回文章
返回