AOD-Net: All-in-One Dehazing Network

被引:1572
作者
Li, Boyi [1 ]
Peng, Xiulian [2 ]
Wang, Zhangyang [3 ]
Xu, Jizheng [2 ]
Feng, Dan [1 ]
机构
[1] Huazhong Univ Sci & Technol, Wuhan Natl Lab Optoelect, Wuhan, Hubei, Peoples R China
[2] Microsoft Res, Beijing, Peoples R China
[3] Texas A&M Univ, Dept Comp Sci & Engn, College Stn, TX 77843 USA
来源
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV) | 2017年
关键词
IMAGE; FRAMEWORK; ALGORITHM; VISION;
D O I
10.1109/ICCV.2017.511
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper proposes an image dehazing model built with a convolutional neural network (CNN), called All-in-One Dehazing Network (AOD-Net). It is designed based on a re-formulated atmospheric scattering model. Instead of estimating the transmission matrix and the atmospheric light separately as most previous models did, AOD-Net directly generates the clean image through a light-weight CNN. Such a novel end-to-end design makes it easy to embed AOD-Net into other deep models, e.g., Faster R-CNN, for improving high-level tasks on hazy images. Experimental results on both synthesized and natural hazy image datasets demonstrate our superior performance than the state-of-the-art in terms of PSNR, SSIM and the subjective visual quality. Furthermore, when concatenating AOD-Net with Faster R-CNN, we witness a large improvement of the object detection performance on hazy images.
引用
收藏
页码:4780 / 4788
页数:9
相关论文
共 32 条
[1]  
[Anonymous], 2013, P INT C MACH LEAR
[2]  
[Anonymous], COMP VIS PATT REC 20
[3]  
[Anonymous], IEEE T PATTERN ANAL
[4]  
[Anonymous], 2017, ARXIV PREPRINT ARXIV
[5]  
[Anonymous], 2007, Tech. Rep
[6]  
Berman D, 2017, IEEE INT CONF COMPUT, P115
[7]   Non-Local Image Dehazing [J].
Berman, Dana ;
Treibitz, Tali ;
Avidan, Shai .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1674-1682
[8]   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
[9]   Dehazing Using Color-Lines [J].
Fattal, Raanan .
ACM TRANSACTIONS ON GRAPHICS, 2014, 34 (01)
[10]   Single image dehazing [J].
Fattal, Raanan .
ACM TRANSACTIONS ON GRAPHICS, 2008, 27 (03)