Underwater object detection algorithm based on feature enhancement and progressive dynamic aggregation strategy

被引:40
作者
Hua, Xia [1 ]
Cui, Xiaopeng [1 ]
Xu, Xinghua [1 ]
Qiu, Shaohua [1 ]
Liang, Yingjie [1 ]
Bao, Xianqiang [1 ]
Li, Zhong [1 ]
机构
[1] Naval Univ Engn, Natl Key Lab Sci & Technol Vessel Integrated Power, Wuhan 430033, Peoples R China
关键词
Underwater image; Dynamic feature fusion; Small object detection; Rapid spatial pyramid pooling; Feature enhancement;
D O I
10.1016/j.patcog.2023.109511
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
To solve the problems that the conventional object detector is hard to extract features and miss detec-tion of small objects when detecting underwater objects due to the noise of underwater environment and the scale change of objects, this paper designs a novel feature enhancement & progressive dynamic aggregation strategy, and proposes a new underwater object detector based on YOLOv5s. Firstly, a fea-ture enhancement gating module is designed to selectively suppress or enhance multi-level features and reduce the interference of underwater complex environment noise on feature fusion. Then, the adjacent feature fusion mechanism and dynamic fusion module are designed to dynamically learn fusion weights and perform multi-level feature fusion progressively, so as to suppress the conflict information in multi -scale feature fusion and prevent small objects from being submerged by the conflict information. At last, a spatial pyramid pool structure (FMSPP) based on the same size quickly mixed pool layer is proposed, which can make the network obtain stronger description ability of texture and contour features, reduce the parameters, and further improve the generalization ability and classification accuracy. The ablation experiments and multi-method comparison experiments on URPC and DUT-USEG data sets prove the effectiveness of the proposed strategy. Compared with the current mainstream detectors, our detector achieves obvious advantages in detection performance and efficiency.(c) 2023 Published by Elsevier Ltd.
引用
收藏
页数:14
相关论文
共 35 条
[1]  
[Anonymous], About Us
[2]   mSODANet: A network for multi-scale object detection in aerial images using hierarchical dilated convolutions * [J].
Chalavadi, Vishnu ;
Jeripothula, Prudviraj ;
Datla, Rajeshreddy ;
Babu, Sobhan Ch ;
Mohan, Krishna C. .
PATTERN RECOGNITION, 2022, 126
[3]   HCE: Hierarchical Context Embedding for Region-Based Object Detection [J].
Chen, Zhao-Min ;
Jin, Xin ;
Zhao, Bo-Rui ;
Zhang, Xiaoqin ;
Guo, Yanwen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :6917-6929
[4]   Camouflaged Object Detection [J].
Fan, Deng-Ping ;
Ji, Ge-Peng ;
Sun, Guolei ;
Cheng, Ming-Ming ;
Shen, Jianbing ;
Shao, Ling .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :2774-2784
[5]   NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detection [J].
Ghiasi, Golnaz ;
Lin, Tsung-Yi ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :7029-7038
[6]   Squeeze-and-Excitation Networks [J].
Hu, Jie ;
Shen, Li ;
Albanie, Samuel ;
Sun, Gang ;
Wu, Enhua .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (08) :2011-2023
[7]   Underwater Biological Detection Based on YOLOv4 Combined with Channel Attention [J].
Li, Aolun ;
Yu, Long ;
Tian, Shengwei .
JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2022, 10 (04)
[8]  
Li X, 2015, OCEANS MTSIEEE C
[9]   Focal Loss for Dense Object Detection [J].
Lin, Tsung-Yi ;
Goyal, Priya ;
Girshick, Ross ;
He, Kaiming ;
Dollar, Piotr .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (02) :318-327
[10]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944