SADnet: Semi-supervised Single Image Dehazing Method Based on an Attention Mechanism

被引:24
作者
Sun, Ziyi [1 ]
Zhang, Yunfeng [1 ]
Bao, Fangxun [2 ]
Wang, Ping [3 ]
Yao, Xunxiang [4 ]
Zhang, Caiming [5 ]
机构
[1] Shandong Univ Finance & Econ, Dept Comp Sci & Technol, Jinan 250014, Peoples R China
[2] Shandong Univ, Dept Math, Jinan 250100, Peoples R China
[3] Quebec Univ, Ecole Technol Suprieure, Dept Software & IT Engn, Montreal, PQ, Canada
[4] Univ Technol Sydney, Fac Engn & Informat Technol, Sydney, NSW, Australia
[5] Shandong Univ, Dept Comp Sci & Technol, Jinan 250101, Peoples R China
基金
中国国家自然科学基金;
关键词
Single image dehazing; practical applications; semi-supervised; attention; deep learning; HAZE;
D O I
10.1145/3478457
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Many real-life tasks such as military reconnaissance and traffic monitoring require high-quality images. However, images acquired in foggy or hazy weather pose obstacles to the implementation of these real-life tasks; consequently, image dehazing is an important research problem. To meet the requirements of practical applications, a single image dehazing algorithm has to be able to effectively process real-world hazy images with high computational efficiency. In this article, we present a fast and robust semi-supervised dehazing algorithm named SADnet for practical applications. SADnet utilizes both synthetic datasets and natural hazy images for training, so it has good generalizability for real-world hazy images. Furthermore, considering the uneven distribution of haze in the atmospheric environment, a Channel-Spatial Self-Attention (CSSA) mechanism is presented to enhance the representational power of the proposed SADnet. Extensive experimental results demonstrate that the presented approach achieves good dehazing performances and competitive running times compared with other state-of-the-art image dehazing algorithms.
引用
收藏
页数:23
相关论文
共 62 条
[1]  
Ancuti CO, 2011, LECT NOTES COMPUT SC, V6493, P501
[2]   Supervised, Unsupervised, and Semi-Supervised Feature Selection: A Review on Gene Selection [J].
Ang, Jun Chin ;
Mirzal, Andri ;
Haron, Habibollah ;
Hamed, Haza Nuzly Abdull .
IEEE-ACM TRANSACTIONS ON COMPUTATIONAL BIOLOGY AND BIOINFORMATICS, 2016, 13 (05) :971-989
[3]   Non-Local Image Dehazing [J].
Berman, Dana ;
Treibitz, Tali ;
Avidan, Shai .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1674-1682
[4]   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
[5]   An Advanced Visibility Restoration Algorithm for Single Hazy Images [J].
Chen, Bo-Hao ;
Huang, Shih-Chia .
ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2015, 11 (04) :1-21
[6]   Gated Context Aggregation Network for Image Dehazing and Deraining [J].
Chen, Dongdong ;
He, Mingming ;
Fan, Qingnan ;
Liao, Jing ;
Zhang, Liheng ;
Hou, Dongdong ;
Yuan, Lu ;
Hua, Gang .
2019 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2019, :1375-1383
[7]   SCA-CNN: Spatial and Channel-wise Attention in Convolutional Networks for Image Captioning [J].
Chen, Long ;
Zhang, Hanwang ;
Xiao, Jun ;
Nie, Liqiang ;
Shao, Jian ;
Liu, Wei ;
Chua, Tat-Seng .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6298-6306
[8]   Discrete Haze Level Dehazing Network [J].
Cong, Xiaofeng ;
Gui, Jie ;
Miao, Kai-Chao ;
Zhang, Jun ;
Wang, Bing ;
Chen, Peng .
MM '20: PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, 2020, :1828-1836
[9]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[10]  
Devlin J, 2019, Arxiv, DOI [arXiv:1810.04805, 10.48550/arXiv.1810.04805]