Learning Deformable and Attentive Network for image restoration

被引:16
作者
Huang, Yuan [1 ]
Hou, Xingsong [1 ]
Dun, Yujie [1 ]
Qin, Jie [2 ]
Liu, Li [3 ]
Qian, Xueming [1 ]
Shao, Ling [3 ]
机构
[1] Xi An Jiao Tong Univ, Sch Informat & Commun Engn, Xian 710049, Peoples R China
[2] Nanjing Univ Aeronaut & Astronaut, Coll Comp Sci & Technol, Nanjing 210016, Peoples R China
[3] Incept Inst Artificial Intelligence, Abu Dhabi, U Arab Emirates
基金
国家重点研发计划;
关键词
Image restoration; Convolution neural network; Deformable convolution; Attention mechanism; Knowledge distillation; Image denoising; JPEG artifacts removal; Real-world super resolution; QUALITY ASSESSMENT;
D O I
10.1016/j.knosys.2021.107384
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Image restoration (IR) aims to recover image quality from various degradations. Existing convolutional neural networks (CNN) based IR methods try to improve performance by enlarging the model receptive field with the sacrifice of fine spatial details and extra artifacts. This paper proposes a Deformable and Attentive Network (DANet) to address these problems. In DANet, we propose two novel blocks: Attentive DEformable-convolution Block (ADEB) and Attentive Recurrent Offset Block (AROB). In ADEB, deformable convolution is collaborated with various attention modules to generate more adaptive receptive fields. AROB transfers more attentive texture information among different scales during the encoding/decoding process for ADEB. To further refine DANet, we propose a knowledge distillation scheme to train a light-weighted DANet (DANet-S) with limited performance loss. Extensive experiments on several image benchmark datasets demonstrate that our method achieves state-of-theart (SOTA) results for various IR tasks, including image denoising, JPEG artifacts removal, and real-world super resolution. (C) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页数:14
相关论文
共 58 条
[1]   A High-Quality Denoising Dataset for Smartphone Cameras [J].
Abdelhamed, Abdelrahman ;
Lin, Stephen ;
Brown, Michael S. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1692-1700
[2]   NTIRE 2017 Challenge on Single Image Super-Resolution: Dataset and Study [J].
Agustsson, Eirikur ;
Timofte, Radu .
2017 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2017, :1122-1131
[3]   Real Image Denoising with Feature Attention [J].
Anwar, Saeed ;
Barnes, Nick .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3155-3164
[4]  
Burger HC, 2012, PROC CVPR IEEE, P2392, DOI 10.1109/CVPR.2012.6247952
[5]   Toward Real-World Single Image Super-Resolution: A New Benchmark and A New Model [J].
Cai, Jianrui ;
Zeng, Hui ;
Yong, Hongwei ;
Cao, Zisheng ;
Zhang, Lei .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3086-3095
[6]  
Caruana R., 2006, P ACM INT C KNOWLEDG, P535, DOI DOI 10.1145/1150402.1150464
[7]  
Chen YJ, 2015, PROC CVPR IEEE, P5261, DOI 10.1109/CVPR.2015.7299163
[8]   Color image denoising via sparse 3d collaborative filtering with grouping constraint in luminance-chrominance space [J].
Dabov, Kostadin ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
2007 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, VOLS 1-7, 2007, :313-316
[9]   Image denoising by sparse 3-D transform-domain collaborative filtering [J].
Dabov, Kostadin ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2007, 16 (08) :2080-2095
[10]   Deformable Convolutional Networks [J].
Dai, Jifeng ;
Qi, Haozhi ;
Xiong, Yuwen ;
Li, Yi ;
Zhang, Guodong ;
Hu, Han ;
Wei, Yichen .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :764-773