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 条
  • [1] A hybrid attention network with convolutional neural network and transformer for underwater image restoration
    Jiao, Zhan
    Wang, Ruizi
    Zhang, Xiangyi
    Fu, Bo
    Thanh, Dang Ngoc Hoang
    PEERJ COMPUTER SCIENCE, 2023, 9
  • [2] Attention-guided hybrid transformer-convolutional neural network for underwater image super-resolution
    Zhan, Zihan
    Li, Chaofeng
    Zhang, Yuqi
    JOURNAL OF ELECTRONIC IMAGING, 2024, 33 (01)
  • [3] Underwater Image Restoration Based on a Parallel Convolutional Neural Network
    Wang, Keyan
    Hu, Yan
    Chen, Jun
    Wu, Xianyun
    Zhao, Xi
    Li, Yunsong
    REMOTE SENSING, 2019, 11 (13)
  • [4] Hybrid network of convolutional neural network and transformer for deepfake geographic image detection
    Liu, Xiaoyong
    Dong, Xiaofei
    Xie, Feng
    Lu, Pei
    Lu, Xi
    Jiang, Mingzhong
    JOURNAL OF ELECTRONIC IMAGING, 2024, 33 (02)
  • [5] UCRNet: Underwater color image restoration via a polarization-guided convolutional neural network
    Hu, Haofeng
    Huang, Yizhao
    Li, Xiaobo
    Jiang, Liubing
    Che, Li
    Liu, Tiegen
    Zhai, Jingsheng
    FRONTIERS IN MARINE SCIENCE, 2022, 9
  • [6] Hybrid Convolutional and Attention Network for Hyperspectral Image Denoising
    Hu, Shuai
    Gao, Feng
    Zhou, Xiaowei
    Dong, Junyu
    Du, Qian
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [7] CANet: Concatenated Attention Neural Network for Image Restoration
    Tian, YingJie
    Wang, Yiqi
    Yang, LinRui
    Qi, ZhiQuan
    IEEE SIGNAL PROCESSING LETTERS, 2020, 27 : 1615 - 1619
  • [8] Canet: Concatenated attention neural network for image restoration
    Tian, YingJie
    Wang, Yiqi
    Yang, LinRui
    Qi, ZhiQuan
    Qi, ZhiQuan (qizhiquan@ucas.ac.cn), 1615, Institute of Electrical and Electronics Engineers Inc. (27): : 1615 - 1619
  • [9] Underwater image restoration for seafloor targets with hybrid attention mechanisms and conditional generative adversarial network
    Yang, Peng
    Wu, Heng
    He, Chunhua
    Luo, Shaojuan
    DIGITAL SIGNAL PROCESSING, 2023, 134
  • [10] Global and pyramid convolutional neural network with hybrid attention mechanism for hyperspectral image classification
    Wu, Linfeng
    Wang, Huajun
    GEOCARTO INTERNATIONAL, 2023, 38 (01)