URNet: A U-Net based residual network for image dehazing

被引:27
作者
Feng, Ting [1 ,2 ]
Wang, Chuansheng [3 ]
Chen, Xinwei [4 ]
Fan, Haoyi [5 ]
Zeng, Kun [1 ]
Li, Zuoyong [1 ]
机构
[1] Minjiang Univ, Coll Comp & Control Engn, Fujian Prov Key Lab Informat Proc & Intelligent C, Fuzhou 350121, Peoples R China
[2] Fuzhou Univ, Coll Math & Comp Sci, Fuzhou 350108, Peoples R China
[3] Chinese Acad Sci, Shenzhen Inst Adv Technol, Shenzhen 440305, Peoples R China
[4] Minjiang Univ, Fujian Engn & Res Ctr New Chinese Lacquer Mat, Fuzhou 350121, Peoples R China
[5] Harbin Univ Sci & Technol, Sch Comp Sci & Technol, Harbin 150080, Peoples R China
基金
中国国家自然科学基金;
关键词
Image dehazing; Deep learning; Feature extraction; SINGLE; ALGORITHM;
D O I
10.1016/j.asoc.2020.106884
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Low visibility in hazy weather causes the loss of image details in digital images captured by some imaging devices such as monitors. This paper proposes an end-to-end U-Net based residual network (URNet) to improve the visibility of hazy images. The encoder module of URNet uses hybrid convolution combining standard convolution with dilated convolution to expand the receptive field for extracting image features with more details. The URNet embeds several building blocks of ResNet into the junction between the encoder module and the decoder module. This prevents network performance degradation due to the vanishing gradient. After considering large absolute difference on image saturation and value components between hazy images and haze-free images in the HSV color space, the URNet defines a new loss function to better guide the network training. Experimental results on synthetic hazy images and real hazy images show that the URNet significantly improves the image dehazing effect compared to the state-of-the-art methods. (C) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:9
相关论文
共 38 条
  • [1] 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
  • [2] Multi-scale Adaptive Dehazing Network
    Chen, Shuxin
    Chen, Yizi
    Qu, Yanyun
    Huang, Jingying
    Hong, Ming
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2019), 2019, : 2051 - 2059
  • [3] RI-GAN: An End-to-End Network for Single Image Haze Removal
    Dudhane, Akshay
    Aulakh, Harshjeet Singh
    Murala, Subrahmanyam
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2019), 2019, : 2014 - 2023
  • [4] Fan HY, 2020, INT CONF ACOUST SPEE, P5685, DOI [10.1109/ICASSP40776.2020.9053387, 10.1109/icassp40776.2020.9053387]
  • [5] Correlation-Aware Deep Generative Model for Unsupervised Anomaly Detection
    Fan, Haoyi
    Zhang, Fengbin
    Wang, Ruidong
    Xi, Liang
    Li, Zuoyong
    [J]. ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PAKDD 2020, PT II, 2020, 12085 : 688 - 700
  • [6] LeukocyteMask: An automated localization and segmentation method for leukocyte in blood smear images using deep neural networks
    Fan, Haoyi
    Zhang, Fengbin
    Xi, Liang
    Li, Zuoyong
    Liu, Guanghai
    Xu, Yong
    [J]. JOURNAL OF BIOPHOTONICS, 2019, 12 (07)
  • [7] Halder S.S., 2018, IEEE T IMAGE PROCESS, DOI DOI 10.1109/TIP.2018.2841198
  • [8] Haoran Xu, 2012, 2012 IEEE International Conference on Information Science and Technology, P663, DOI 10.1109/ICIST.2012.6221729
  • [9] Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1026 - 1034
  • [10] 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