A hybrid attention network with convolutional neural network and transformer for underwater image restoration

被引:0
|
作者
Jiao Z. [1 ]
Wang R. [1 ]
Zhang X. [2 ]
Fu B. [2 ]
Thanh D.N.H. [3 ]
机构
[1] Liaoning Vocational College of Light Industry, Liaoning, Dalian
[2] Liaoning Normal University, Liaoning, Dalian
[3] Department of Information Technology, College of Technology and Design, University of Economics Ho Chi, Minh City, Ho Chi Minh City
基金
中国博士后科学基金;
关键词
Attention mechanism; Convolutional neural network; Deep learning; Hybrid attention; Image processing; Transformer; Underwater image restoration;
D O I
10.7717/PEERJ-CS.1559
中图分类号
学科分类号
摘要
The analysis and communication of underwater images are often impeded by various elements such as blur, color cast, and noise. Existing restoration methods only address specific degradation factors and struggle with complex degraded images. Furthermore, traditional convolutional neural network (CNN) based approaches may only restore local color while ignoring global features. The proposed hybrid attention network combining CNN and Transformer focuses on addressing these issues. CNN captures local features and the Transformer uses multi-head self-attention to model global relationships. The network also incorporates degraded channel attention and supervised attention mechanisms to refine relevant features and correlations. The proposed method fared better than existing methods in a variety of qualitative criteria when evaluated against the public EUVP dataset of underwater images. © 2023 Jiao et al.
引用
收藏
相关论文
共 50 条
  • [21] A Visual Attention Based Convolutional Neural Network for Image Classification
    Chen, Yaran
    Zhao, Dongbin
    Lv, Le
    Li, Chengdong
    PROCEEDINGS OF THE 2016 12TH WORLD CONGRESS ON INTELLIGENT CONTROL AND AUTOMATION (WCICA), 2016, : 764 - 769
  • [22] Completed attention convolutional neural network for MRI image segmentation
    Zhang, Zhong
    Lv, Shijie
    Liu, Shuang
    Xiao, Baihua
    High Technology Letters, 2022, 28 (03) : 247 - 251
  • [23] Method of convolutional neural network with hybrid attention for crack detection
    Qu, Zhong
    Chen, Shuwei
    Li, Yanxin
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (05)
  • [24] Dual convolutional neural network with attention for image blind denoising
    Wu, Wencong
    Lv, Guannan
    Duan, Yingying
    Liang, Peng
    Zhang, Yungang
    Xia, Yuelong
    MULTIMEDIA SYSTEMS, 2024, 30 (05)
  • [25] Completed attention convolutional neural network for MRI image segmentation
    张重
    LV Shijie
    LIU Shuang
    XIAO Baihua
    HighTechnologyLetters, 2022, 28 (03) : 247 - 251
  • [26] A Convolutional Neural Network for Brain CT Image Classification Based on Residual Hybrid Attention Mechanism
    基于残差混合注意力机制的脑部CT图像分类卷积神经网络模型
    Pang, Shan-Chen (pangsc@upc.edu.cn), 1600, Chinese Institute of Electronics (49): : 984 - 991
  • [27] A Hybrid Residual Attention Convolutional Neural Network for Compressed Sensing Magnetic Resonance Image Reconstruction
    Hossain, Md. Biddut
    Kwon, Ki-Chul
    Shinde, Rupali Kiran
    Imtiaz, Shariar Md
    Kim, Nam
    DIAGNOSTICS, 2023, 13 (07)
  • [28] Cformer: An underwater image enhancement hybrid network combining convolution and transformer
    Deng, Ruhui
    Zhao, Lei
    Li, Heng
    Liu, Hui
    IET IMAGE PROCESSING, 2023, 17 (13) : 3841 - 3855
  • [29] Underwater Image Enhancement Based on Dense Cascaded Convolutional Neural Network
    Chen Qingjiang
    Xie Yali
    LASER & OPTOELECTRONICS PROGRESS, 2022, 59 (22)
  • [30] Sonar image recognition of underwater target based on convolutional neural network
    Jin, Leilei
    Liang, Hong
    Yang, Changsheng
    Xibei Gongye Daxue Xuebao/Journal of Northwestern Polytechnical University, 2021, 39 (02): : 285 - 291