Laser & Optoelectronics Progress, Volume. 58, Issue 14, 1410017(2021)

Feature Constraint CycleGAN for Single Image Dehazing Algorithm

Dianwei Wang1, Shunli Li1、*, Pengfei Han2, Ying Liu1, Jing Jiang1, and Xincheng Ren3
Author Affiliations
  • 1School of Communications and Information Engineering, Xi'an University of Posts and Telecommunications, Xi'an, Shaanxi 710121, China
  • 2Center for AI Research and Innovation, Westlake University, Hangzhou, Zhejiang 310024, China
  • 3School of Physics and Electronic Information, Yan'an University, Yan'an, Shaanxi 716000, China
  • show less

    To solve the problem of serious degradation of images captured by imaging equipment in hazy scenes, a single image dehazing algorithm based on feature constraint cycle generative adversarial networks (CycleGAN) is proposed. First, the mapping relationship between the hazy image and the clear image is learned by the CycleGAN, and the discriminator determines whether the reconstructed image conforms to the data distribution of the real sample image. Then, as for network model loss function, the joint function based on cyclic consistent loss and Haze loss is constructed. The frequency information of image is introduced into Haze loss function as a constraint term and the content loss of high-dimensional feature is extracted by the pre-trained VGG-16 model, which can improve dehazing performance of the network and solve the problems of insufficient dehazing and image information loss in CycleGAN model. Finally, the peak signal-to-noise ratio and structural similarity index are used as the evaluation criteria for quantitative analysis between the input hazy image and the output dehazing enhanced image. Experimental results show that the proposed algorithm can effectively reduce the effect of haze on imaging quality, and obtain better subjective visual evaluation and objective quantitative evaluation.

    Tools

    Get Citation

    Copy Citation Text

    Dianwei Wang, Shunli Li, Pengfei Han, Ying Liu, Jing Jiang, Xincheng Ren. Feature Constraint CycleGAN for Single Image Dehazing Algorithm[J]. Laser & Optoelectronics Progress, 2021, 58(14): 1410017

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Sep. 21, 2020

    Accepted: Nov. 18, 2020

    Published Online: Jun. 30, 2021

    The Author Email: Li Shunli (780953701@qq.com)

    DOI:10.3788/LOP202158.1410017

    Topics