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 条
  • [1] Lightweight Fire Detection Algorithm Based on Multi-Scale Feature Fusion
    Yang, Guowei
    Liu, Xuan
    Gao, Min
    Xu, Di
    Computer Engineering and Applications, 60 (23): : 229 - 237
  • [2] A multi-scale feature fusion target detection algorithm
    Dong, Chong
    Li, Jingmei
    Wang, Jiaxiang
    2018 INTERNATIONAL CONFERENCE ON IMAGE AND VIDEO PROCESSING, AND ARTIFICIAL INTELLIGENCE, 2018, 10836
  • [3] Lightweight Face Detection Algorithm with Multi-scale Feature Fusion
    Wang J.
    Song X.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2022, 35 (06): : 507 - 515
  • [4] Drone Target Detection Algorithm Based on Multi-scale Fusion and Lightweight Network
    Xue S.
    Lu T.
    Lü Q.
    Cao G.
    Hunan Daxue Xuebao/Journal of Hunan University Natural Sciences, 2023, 50 (08): : 82 - 93
  • [5] LIGHTWEIGHT SALIENCY TARGET INTELLIGENT DETECTION BASED ON MULTI-SCALE FEATURE ADAPTIVE FUSION
    Zhu, Muqing
    SCALABLE COMPUTING-PRACTICE AND EXPERIENCE, 2024, 25 (02): : 883 - 890
  • [6] A Lightweight Detection Algorithm for Unmanned Surface Vehicles Based on Multi-Scale Feature Fusion
    Zhang, Lei
    Du, Xiang
    Zhang, Renran
    Zhang, Jian
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (07)
  • [7] Deep Multi-Scale Feature Fusion Target Detection Algorithm Based on Deep Learning
    Liu Xin
    Chen Siyi
    Chen Xiaolong
    Du Xinhao
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (12)
  • [8] Underwater image object detection based on multi-scale feature fusion
    Yang, Chao
    Zhang, Ce
    Jiang, Longyu
    Zhang, Xinwen
    MACHINE VISION AND APPLICATIONS, 2024, 35 (06)
  • [9] The Underwater Target Detection Based on Multi-Feature Fusion Algorithm
    Xu Zhijing
    Cao Peipei
    2011 INTERNATIONAL CONFERENCE ON COMPUTERS, COMMUNICATIONS, CONTROL AND AUTOMATION (CCCA 2011), VOL II, 2010, : 460 - 463
  • [10] A Multi-Scale Feature Fusion Based Lightweight Vehicle Target Detection Network on Aerial Optical Images
    Yu, Chengrui
    Jiang, Xiaonan
    Wu, Fanlu
    Fu, Yao
    Pei, Junyan
    Zhang, Yu
    Li, Xiangzhi
    Fu, Tianjiao
    REMOTE SENSING, 2024, 16 (19)