Single Image Dehazing Using CycleGAN Based on Feature Fusion

被引:0
作者
Jin, Xiaofei [1 ]
Zhang, Denyin [2 ]
Lu, Songhao [2 ]
Guo, Dingxu [2 ]
Ni, Wenye [1 ]
Li, Xu [1 ]
机构
[1] Nanjing Univ Posts & Telecommun, Sch Commun & Informat Engn, Nanjing, Peoples R China
[2] Nanjing Univ Posts & Telecommun, Sch Internet Things, Nanjing, Peoples R China
来源
2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC | 2023年
基金
中国国家自然科学基金;
关键词
Image Dehazing; CycleGAN; Feature Fusion; Dense Residual Blocks; Attention Mechanisms;
D O I
10.1109/CCDC58219.2023.10327278
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Due to the difficulty of obtaining paired data sets from the real world to train the network, most of the current dehazing networks arc trained by synthetic hazy data sets, which will have drawbacks such as poor generalization ability to natural haze scenes and loss of depth details. This paper proposes an image dehazing method using CycleGAN based on improved feature fusion to solve the problem. The method is designed with an encoder-decoder structure in the generator network, enabling more feature information to be extracted at multiple scales. In order to restore the detailed information of the image, this paper introduces the residual dense block instead of the convolution module to extract and fuse the feature information under different receptive Fields in each stage of the network. Aiming at the complexity of the fog distribution in the actual scene, this paper introduces an improved channel and spatial attention mechanism in the skip connection of the network to accomplish non-uniform processing of haze areas with different concentrations. At the same time, to improve the quality of the generated image, this paper introduces perceptual loss to enhance the detailed information of the output features, making the generated image more realistic. The experimental findings suggest that the proposed method can achieve better subjective visual effects and image details, and the outcomes of objective indicators are also improved.
引用
收藏
页码:1642 / 1648
页数:7
相关论文
共 25 条
  • [1] A survey on analysis and implementation of state-of-the-art haze removal techniques
    Babu, G. Harish
    Venkatram, N.
    [J]. JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2020, 72
  • [2] DehazeNet: An End-to-End System for Single Image Haze Removal
    Cai, Bolun
    Xu, Xiangmin
    Jia, Kui
    Qing, Chunmei
    Tao, Dacheng
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (11) : 5187 - 5198
  • [3] Gated Context Aggregation Network for Image Dehazing and Deraining
    Chen, Dongdong
    He, Mingming
    Fan, Qingnan
    Liao, Jing
    Zhang, Liheng
    Hou, Dongdong
    Yuan, Lu
    Hua, Gang
    [J]. 2019 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2019, : 1375 - 1383
  • [4] Chen X, 2020, P AS C COMP VIS
  • [5] Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing
    Engin, Deniz
    Genc, Anil
    Ekenel, Hazim Kemal
    [J]. PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, : 938 - 946
  • [6] Single Image Haze Removal Using Dark Channel Prior
    He, Kaiming
    Sun, Jian
    Tang, Xiaoou
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (12) : 2341 - 2353
  • [7] Novel multi-scale retinex with color restoration on graphics processing unit
    Jiang, Bo
    Woodell, Glenn A.
    Jobson, Daniel J.
    [J]. JOURNAL OF REAL-TIME IMAGE PROCESSING, 2015, 10 (02) : 239 - 253
  • [8] Benchmarking Single-Image Dehazing and Beyond
    Li, Boyi
    Ren, Wenqi
    Fu, Dengpan
    Tao, Dacheng
    Feng, Dan
    Zeng, Wenjun
    Wang, Zhangyang
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (01) : 492 - 505
  • [9] AOD-Net: All-in-One Dehazing Network
    Li, Boyi
    Peng, Xiulian
    Wang, Zhangyang
    Xu, Jizheng
    Feng, Dan
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 4780 - 4788
  • [10] Single Image Dehazing via Conditional Generative Adversarial Network
    Li, Runde
    Pan, Jinshan
    Li, Zechao
    Tang, Jinhui
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 8202 - 8211