Semantic attention and relative scene depth-guided network for underwater image enhancement

被引:19
作者
Chen, Tingkai [1 ]
Wang, Ning [2 ]
Chen, Yanzheng [2 ]
Kong, Xiangjun [1 ]
Lin, Yejin [2 ]
Zhao, Hong [1 ]
Karimi, Hamid Reza [3 ]
机构
[1] Dalian Maritime Univ, Sch Marine Elect Engn, Dalian 116026, Peoples R China
[2] Dalian Maritime Univ, Sch Marine Engn, Dalian 116026, Peoples R China
[3] Politecn Milan, Dept Mech Engn, I-20156 Milan, Italy
基金
中国国家自然科学基金;
关键词
Underwater image enhancement; Multi-color space feature representation; network; Underwater relative scene depth estimation; Underwater scene semantic segmentation;
D O I
10.1016/j.engappai.2023.106532
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this paper, to solve unique underwater degradation challenges covering low contrast, color deviation and blurring, etc., a novel semantic attention and relative scene depth-guided network (SARSDN) for underwater im-age enhancement is proposed. Main contributions are as follows: (1) By combining with diverse characteristics of red-green-blue, hue-saturation-value and Lab spaces, the multi-color space feature representation network (MSFRN) is elaborately developed, such that domain shifting can be effectively alleviated; (2) By utilizing position attention and devising multi-dilated-convolution depth perception unit, the underwater relative scene depth estimation network (URSDEN) is proposed to adapt attention weights to regions with different degrees of degradation, thereby exclusively accommodating scene depth-dependent attenuation and scattering; (3) The underwater scene semantic segmentation network (USSSN) is devised to estimate semantic attention map for reducing artifacts and increasing integrity of foreground objects during underwater image enhancement by virtue of encoder-decoder framework with deformable convolution network; and (4) The entire SARSDN scheme is ultimately created in a modular manner by integrating MSFRN, URSDEN and USSSN modules. Comprehensive experiments and comparisons thoroughly illustrate that the developed SARSDN framework outperforms typical underwater image enhancement approaches from both subjective and objective aspects, where UIQM scores are 0.8263, 0.9393, 1.1817, 0.5289, 0.6517, 0.5393, 0.7917 and 0.4651 higher than those of IBLA, ULAP, HLRP, UCM, RGHS, MLLE, UGAN and FUnIE-GAN schemes, respectively.
引用
收藏
页数:18
相关论文
共 74 条
[1]   SFE: A Simple, Fast, and Efficient Feature Selection Algorithm for High-Dimensional Data [J].
Ahadzadeh, Behrouz ;
Abdar, Moloud ;
Safara, Fatemeh ;
Khosravi, Abbas ;
Menhaj, Mohammad Bagher ;
Suganthan, Ponnuthurai Nagaratnam .
IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2023, 27 (06) :1896-1911
[2]   A Revised Underwater Image Formation Model [J].
Akkaynak, Derya ;
Treibitz, Tali .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6723-6732
[3]   Compared-neighborhood based image dehazing for improved visibility [J].
Alenezi, Fayadh .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 121
[4]   Underwater image dehazing using global color features [J].
Alenezi, Fayadh ;
Armghan, Ammar ;
Santosh, K. C. .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 116
[5]   Effective Single Image Dehazing by Fusion [J].
Ancuti, Codruta Orniana ;
Ancuti, Cosmin ;
Bekaert, Philippe .
2010 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, 2010, :3541-3544
[6]  
Ancuti C, 2012, PROC CVPR IEEE, P81, DOI 10.1109/CVPR.2012.6247661
[7]   Underwater Single Image Color Restoration Using Haze-Lines and a New Quantitative Dataset [J].
Berman, Dana ;
Levy, Deborah ;
Avidan, Shai ;
Treibitz, Tali .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2021, 43 (08) :2822-2837
[8]   CURE-Net: A Cascaded Deep Network for Underwater Image Enhancement [J].
Cai, Xiaowen ;
Jiang, Nanfeng ;
Chen, Weiling ;
Hu, Jinsong ;
Zhao, Tiesong .
IEEE JOURNAL OF OCEANIC ENGINEERING, 2024, 49 (01) :226-236
[10]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851