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 条
  • [41] MacNet: a mobile attention classification network combining convolutional neural network and transformer for the differentiation of cervical cancer
    An, Yi
    Lei, Yuanyuan
    Huang, Zhenxing
    Liu, Yu
    Huang, Meiyong
    Liu, Zhou
    Li, Wenbo
    Liang, Dong
    Huang, Wenting
    Hu, Zhanli
    QUANTITATIVE IMAGING IN MEDICINE AND SURGERY, 2025, 15 (01) : 55 - 73
  • [42] A Lightweight Hybrid Convolutional Neural Network for Hyperspectral Image Classification
    Ma, Xiaohu
    Kang, Xudong
    Qin, Huawei
    Wang, Wuli
    Ren, Guangbo
    Wang, Jianbu
    Liu, Baodi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [43] HCNNet: A Hybrid Convolutional Neural Network for Spatiotemporal Image Fusion
    Zhu, Zhuangshan
    Tao, Yuxiang
    Luo, Xiaobo
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [44] A hybrid enhanced attention transformer network for medical ultrasound image segmentation
    Jiang, Tao
    Xing, Wenyu
    Yu, Ming
    Ta, Dean
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 86
  • [45] Pyramid Attention Network for Image Restoration
    Yiqun Mei
    Yuchen Fan
    Yulun Zhang
    Jiahui Yu
    Yuqian Zhou
    Ding Liu
    Yun Fu
    Thomas S. Huang
    Humphrey Shi
    International Journal of Computer Vision, 2023, 131 : 3207 - 3225
  • [46] Pyramid Attention Network for Image Restoration
    Mei, Yiqun
    Fan, Yuchen
    Zhang, Yulun
    Yu, Jiahui
    Zhou, Yuqian
    Liu, Ding
    Fu, Yun
    Huang, Thomas S.
    Shi, Humphrey
    INTERNATIONAL JOURNAL OF COMPUTER VISION, 2023, 131 (12) : 3207 - 3225
  • [47] Attention Cube Network for Image Restoration
    Hang, Yucheng
    Liao, Qingmin
    Yang, Wenming
    Chen, Yupeng
    Zhou, Jie
    MM '20: PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, 2020, : 2562 - 2570
  • [48] Lightweight Attention Convolutional Neural Network for Retinal Vessel Image Segmentation
    Li, Xiang
    Jiang, Yuchen
    Li, Minglei
    Yin, Shen
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2021, 17 (03) : 1958 - 1967
  • [49] Hyperspectral Image Classification Based on Dilated Convolutional Attention Neural Network
    Zhang Xiangdong
    Wang Tengjun
    Zhu Shaojun
    Yang Yun
    ACTA OPTICA SINICA, 2021, 41 (03)
  • [50] EEG Classification Using Hybrid Convolutional Neural Network with Attention Mechanism
    Ciurea, Alexe
    Manoila, Cristina-Petruta
    Ionescu, Bogdan
    ADVANCES IN DIGITAL HEALTH AND MEDICAL BIOENGINEERING, VOL 1, EHB-2023, 2024, 109 : 783 - 791