Underwater Image Enhancement Using Encoder-Decoder Scale Attention Network

被引:0
作者
Lee, Ka-Ki [1 ]
Hsieh, Jun-Wei [1 ]
Hsieh, Yi-Kuan [1 ]
Hsieh, An-Ting [2 ]
机构
[1] Natl Yang Ming Chiao Tung Univ, Inst Computat Intelligence, Tainan, Taiwan
[2] Natl Tsing Hua Univ, Inst Informat Syst & Applicat, Hsinchu, Taiwan
来源
2024 6TH INTERNATIONAL CONFERENCE ON COMPUTER COMMUNICATION AND THE INTERNET, ICCCI 2024 | 2024年
关键词
Deep learning; Underwater image enhancement; EDSA-Net; Underwater image deblurring;
D O I
10.1109/ICCCI62159.2024.10674352
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Capturing underwater images often results in color distortions and blurriness due to the influence of water, leading to color shifts and haze caused by light propagation. To tackle these issues, we introduce the Encoder-Decoder Scale Attention Network (EDSA-Net) in this paper. The EDSA-Net integrates a Scale-Attention Module (SAM) to facilitate channel weight calculation across scales, thus extracting more informative features from the backbone. Additionally, we introduce a novel Cross-Scale Attention Module (CSAM) to address scale inconsistency and selectively weigh feature maps across scales, capturing various features crucial for underwater image restoration. By enabling interactions among different scales, the network extracts essential features from the images and progressively decodes them to restore clear and sharp underwater images. Experimental results demonstrate that our method achieves state-of-the-art (SoTA) performance and surpasses existing approaches in the field of underwater image restoration.
引用
收藏
页码:101 / 106
页数:6
相关论文
共 19 条
  • [1] Transmission Estimation in Underwater Single Images
    Drews-, P., Jr.
    do Nascimento, E.
    Moraes, F.
    Botelho, S.
    Campos, M.
    [J]. 2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2013, : 825 - 830
  • [2] Fabbri C, 2018, IEEE INT CONF ROBOT, P7159
  • [3] Fu XY, 2014, IEEE IMAGE PROC, P4572, DOI 10.1109/ICIP.2014.7025927
  • [4] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [5] DSLR-Quality Photos on Mobile Devices with Deep Convolutional Networks
    Ignatov, Andrey
    Kobyshev, Nikolay
    Timofte, Radu
    Vanhoey, Kenneth
    Luc Van Gool
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 3297 - 3305
  • [6] Iqbal K, 2010, IEEE INT C SYSTEMS M
  • [7] Fast Underwater Image Enhancement for Improved Visual Perception
    Islam, Md Jahidul
    Xia, Youya
    Sattar, Junaed
    [J]. IEEE ROBOTICS AND AUTOMATION LETTERS, 2020, 5 (02) : 3227 - 3234
  • [8] Underwater Image Enhancement by Dehazing With Minimum Information Loss and Histogram Distribution Prior
    Li, Chong-Yi
    Guo, Ji-Chang
    Cong, Run-Min
    Pang, Yan-Wei
    Wang, Bo
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 26 (12) : 5664 - 5677
  • [9] Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embedding
    Li, Chongyi
    Anwar, Saeed
    Hou, Junhui
    Cong, Runmin
    Guo, Chunle
    Ren, Wenqi
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 : 4985 - 5000
  • [10] An Underwater Image Enhancement Benchmark Dataset and Beyond
    Li, Chongyi
    Guo, Chunle
    Ren, Wenqi
    Cong, Runmin
    Hou, Junhui
    Kwong, Sam
    Tao, Dacheng
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 4376 - 4389