Underwater Target Detection Lightweight Algorithm Based on Multi-Scale Feature Fusion

被引:14
|
作者
Chen, Liang [1 ]
Yang, Yuyi [1 ]
Wang, Zhenheng [1 ]
Zhang, Jian [1 ]
Zhou, Shaowu [1 ]
Wu, Lianghong [1 ]
机构
[1] Hunan Univ Sci & Technol, Sch Informat & Elect Engn, Xiangtan 411201, Peoples R China
基金
中国国家自然科学基金;
关键词
underwater target detection; multi-scale fusion; transformer; YOLOv5; lightweight;
D O I
10.3390/jmse11020320
中图分类号
U6 [水路运输]; P75 [海洋工程];
学科分类号
0814 ; 081505 ; 0824 ; 082401 ;
摘要
The performance of underwater target detection algorithms is affected by poor imaging quality in underwater environments. Due to the arithmetic power limitation of underwater devices, existing deep learning networks are unable to provide efficient detection processes with high detection accuracy. Lightweight CNN models have been actively applied for underwater environment detection, yet their lite feature fusion networks cannot provide effective fusion effects and reduce the detection accuracy. In this paper, a lightweight algorithm based on multi-scale feature fusion was proposed, with the model parameters greatly reduced, improving the target detection accuracy. The forward propagation memory overhead is reduced by using multi-scale shared convolutional kernels and pooling operations to co-construct the query matrix in the Tansformer encoding stage. Then, the feature fusion path is optimized in order to enhance the connection of multi-scale features. A multiscale feature adaptive fusion strategy is used to enhance the detection performance and reduce the dependence on the complex feature extraction network. The feature extraction network is also reparameterized to simplify the operation. Using the UPRC offshore dataset for validation, the study results have demonstrated that the statistical mAP metrics validate the detection accuracy. Compared with SSD, RetinaNet and YOLOv5-s improved by 13%, 8.6%, and 0.8%, while the number of parameters decreased by 76.09%, 89.74%, and 87.67%. In addition, compared with the YOLOv5-lite model algorithm with the same parameter volume, the mAP is improved by 3.8%, which verifies the accuracy and efficiency of the algorithm in this paper.
引用
收藏
页数:17
相关论文
共 50 条
  • [31] Multi-scale convolution target detection algorithm with feature pyramid
    Lin Z.-J.
    Luo Z.
    Zhao L.
    Lu D.-M.
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2019, 53 (03): : 533 - 540
  • [32] Drone Detection Based on Multi-scale Feature Fusion
    Zeng, Zhenni
    Wang, Zhenning
    Qin, Lang
    Li, Hui
    2021 6TH INTERNATIONAL CONFERENCE ON UK-CHINA EMERGING TECHNOLOGIES (UCET 2021), 2021, : 194 - 198
  • [33] Light-YOLO: a lightweight detection algorithm based on multi-scale feature enhancement for infrared small ship target
    Tang, Ji
    Hu, Xiao-Min
    Jeon, Sang-Woon
    Chen, Wei-Neng
    COMPLEX & INTELLIGENT SYSTEMS, 2025, 11 (02)
  • [34] LFF-YOLO: A YOLO Algorithm With Lightweight Feature Fusion Network for Multi-Scale Defect Detection
    Qian, Xiaohong
    Wang, Xu
    Yang, Shengying
    Lei, Jingsheng
    IEEE ACCESS, 2022, 10 : 130339 - 130349
  • [35] Lightweight road extraction model based on multi-scale feature fusion
    Liu Y.
    Chen Y.
    Gao L.
    Hong J.
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2024, 58 (05): : 951 - 959
  • [36] MAL-YOLO: a lightweight algorithm for target detection in side-scan sonar images based on multi-scale feature fusion and attention mechanism
    Cao, Yu
    Cui, Xiaodong
    Gan, Mingyi
    Wang, Yaxue
    Yang, Fanlin
    Huang, Yi
    INTERNATIONAL JOURNAL OF DIGITAL EARTH, 2024, 17 (01)
  • [37] Spatial Small Target Detection Method Based on Multi-Scale Feature Fusion Pyramid
    Wang, Xiaojuan
    Liu, Yuepeng
    Xu, Haitao
    Xue, Changbin
    APPLIED SCIENCES-BASEL, 2024, 14 (13):
  • [38] Adaptive feature fusion with attention mechanism for multi-scale target detection
    Ju, Moran
    Luo, Jiangning
    Wang, Zhongbo
    Luo, Haibo
    NEURAL COMPUTING & APPLICATIONS, 2021, 33 (07): : 2769 - 2781
  • [39] Adaptive feature fusion with attention mechanism for multi-scale target detection
    Moran Ju
    Jiangning Luo
    Zhongbo Wang
    Haibo Luo
    Neural Computing and Applications, 2021, 33 : 2769 - 2781
  • [40] Lightweight Target Detection Algorithm Based on Adaptive Spatial Feature Fusion
    Luo Yujie
    Zhang Jian
    Chen Liang
    Zhang Lu
    Ouyang Wanqing
    Huang Daiqin
    Yang Yuyi
    LASER & OPTOELECTRONICS PROGRESS, 2022, 59 (04)