Light-DehazeNet: A Novel Lightweight CNN Architecture for Single Image Dehazing

被引:120
作者
Ullah, Hayat [1 ]
Muhammad, Khan [2 ,3 ]
Irfan, Muhammad [1 ]
Anwar, Saeed [4 ,5 ,6 ]
Sajjad, Muhammad [7 ]
Imran, Ali Shariq [7 ]
de Albuquerque, Victor Hugo C. [8 ]
机构
[1] Sejong Univ, Dept Software, Visual Analyt Knowledge Lab VIS2KNOW Lab, Seoul 143747, South Korea
[2] Sejong Univ, Dept Software, Seoul 143747, South Korea
[3] Sungkyunkwan Univ, Sch Convergence, Coll Comp & Informat, Visual Analyt Knowledge Lab VIS2KNOW Lab, Seoul 03063, South Korea
[4] Data61 Commonwealth Sci & Ind Res Org CSIRO, Black Mt, ACT 2601, Australia
[5] Australian Natl Univ, Res Sch Engn, Canberra, ACT 2601, Australia
[6] Univ Technol Sydney, Sch Comp Sci, Ultimo, NSW 2007, Australia
[7] Norwegian Univ Sci & Technol NTNU, Norwegian Colour & Visual Comp Lab, Dept Comp Sci, N-2815 Gjovik, Norway
[8] Univ Fed Ceara, Dept Teleinformat Engn, BR-60455970 Fortaleza, Ceara, Brazil
关键词
Atmospheric modeling; Image color analysis; Image reconstruction; Computational modeling; Visualization; Image restoration; Atmospheric measurements; Image enhancement; image reconstruction; haze removal; convolutional neural network; single image dehazing; image visibility restoration;
D O I
10.1109/TIP.2021.3116790
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Due to the rapid development of artificial intelligence technology, industrial sectors are revolutionizing in automation, reliability, and robustness, thereby significantly increasing quality and productivity. Most of the surveillance and industrial sectors are monitored by visual sensor networks capturing different surrounding environment images. However, during tempestuous weather conditions, the visual quality of the images is reduced due to contaminated suspended atmospheric particles that affect the overall surveillance systems. To tackle these challenges, this article presents a computationally efficient lightweight convolutional neural network referred to as Light-DehazeNet (LD-Net) for the reconstruction of hazy images. Unlike other learning-based approaches, which separately measure the transmission map and the atmospheric light, our proposed LD-Net jointly estimates both the transmission map and the atmospheric light using a transformed atmospheric scattering model. Furthermore, a color visibility restoration method is proposed to evade the color distortion in the dehaze image. Finally, we conduct extensive experiments using synthetic and natural hazy images. The quantitative and qualitative evaluation on different benchmark hazy datasets verify the superiority of the proposed method over other state-of-the-art image dehazing techniques. Moreover, additional experimentation validates the applicability of the proposed method in the object detection tasks. Considering the lightweight architecture with minimal computational cost, the proposed system is encouraged to be incorporated as an integral part of the vision-based monitoring systems to improve the overall performance.
引用
收藏
页码:8968 / 8982
页数:15
相关论文
共 58 条
[1]   NTIRE 2018 Challenge on Image Dehazing: Methods and Results [J].
Ancuti, Cosmin ;
Ancuti, Codruta O. ;
Timofte, Radu ;
Van Gool, Luc ;
Zhang, Lei ;
Yang, Ming-Hsuan ;
Patel, Vishal M. ;
Zhang, He ;
Sindagi, Vishwanath A. ;
Zhao, Ruhao ;
Ma, Xiaoping ;
Qin, Yong ;
Jia, Limin ;
Friedel, Klaus ;
Ki, Sehwan ;
Sim, Hyeonjun ;
Choi, Jae-Seok ;
Kim, Soo Ye ;
Seo, Soomin ;
Kim, Saehun ;
Kim, Munchurl ;
Mondal, Ranjan ;
Santra, Sanchayan ;
Chanda, Bhabatosh ;
Liu, Jinlin ;
Mei, Kangfu ;
Li, Juncheng ;
Luyao ;
Fang, Faming ;
Jiang, Aiwen ;
Qu, Xiaochao ;
Liu, Ting ;
Wang, Pengfei ;
Sun, Biao ;
Deng, Jiangfan ;
Zhao, Yuhang ;
Hong, Ming ;
Huang, Jingying ;
Chen, Yizhi ;
Chen, Erin ;
Yu, Xiaoli ;
Wu, Tingting ;
Genc, Anil ;
Engin, Deniz ;
Ekenel, Hazim Kemal ;
Liu, Wenzhe ;
Tong, Tong ;
Li, Gen ;
Gao, Qinquan ;
Li, Zhan .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :1004-1014
[2]   Non-Local Image Dehazing [J].
Berman, Dana ;
Treibitz, Tali ;
Avidan, Shai .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1674-1682
[3]   High-Resolution Single Image Dehazing using Encoder-Decoder Architecture [J].
Bianco, Simone ;
Celona, Luigi ;
Piccoli, Flavio ;
Schettini, Raimondo .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2019), 2019, :1927-1935
[4]   Single Image Dehazing Using Color Ellipsoid Prior [J].
Bui, Trung Minh ;
Kim, Wonha .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2018, 27 (02) :999-1009
[5]   DehazeNet: An End-to-End System for Single Image Haze Removal [J].
Cai, Bolun ;
Xu, Xiangmin ;
Jia, Kui ;
Qing, Chunmei ;
Tao, Dacheng .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (11) :5187-5198
[6]   Learning a Single Tucker Decomposition Network for Lossy Image Compression With Multiple Bits-per-Pixel Rates [J].
Cai, Jianrui ;
Cao, Zisheng ;
Zhang, Lei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :3612-3625
[7]   RYF-Net: Deep Fusion Network for Single Image Haze Removal [J].
Dudhane, Akshay ;
Murala, Subrahmanyam .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :628-640
[8]   Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing [J].
Engin, Deniz ;
Genc, Anil ;
Ekenel, Hazim Kemal .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :938-946
[9]   Unsupervised Single Image Dehazing Using Dark Channel Prior Loss [J].
Golts, Alona ;
Freedman, Daniel ;
Elad, Michael .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :2692-2701
[10]   Single Image Haze Removal Using Dark Channel Prior [J].
He, Kaiming ;
Sun, Jian ;
Tang, Xiaoou .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (12) :2341-2353