Underwater-Yolo: Underwater Object Detection Network with Dilated Deformable Convolutions and Dual-Branch Occlusion Attention Mechanism
Underwater object detection is critical for marine ecological monitoring and biodiversity research, yet existing algorithms struggle in detecting densely packed objects of varying sizes, particularly in occluded and complex underwater environments. This study introduces Underwater-Yolo, a novel detection network that enhances performance in these challenging scenarios by integrating a dual-branch occlusion-handling attention mechanism (GLOAM) and a Cross-Stage Partial Dilated Deformable Convolution (CSP-DDC) backbone. The dilated deformable convolutions (DDCs) in the backbone and neck expand the receptive field, thereby improving the detection of small objects, while the deformable convolutions enhance the model’s adaptive feature extraction capabilities for unstructured objects. Additionally, the CARAFE up-sampling operator in the neck aggregates contextual information across a broader spatial domain. The GLOAM, consisting of a global branch (using a Vision Transformer to capture global features and object–background relationships) and a local branch (enhancing the detection of occluded objects through depthwise–pointwise convolutions), further optimizes performance. By incorporating these innovations, the model effectively addresses the challenges of detecting small and occluded objects in dense underwater environments. The evaluation on the CLfish-V1 dataset shows significant improvements over state-of-the-art algorithms, with an AP50 of 93.8%, an AP75 of 88.9%, and an AP-small of 76.4%, marking gains of 4.7%, 16.7%, and 6%, respectively. These results demonstrate the model’s effectiveness in complex underwater scenarios.
Authors
- Wenbo Zhu (ORCID: https://orcid.org/0000-0001-7835-4990)
- Weijie Fu (ORCID: https://orcid.org/0000-0003-2392-3951)
- Yunzhi Zhang (ORCID: https://orcid.org/0000-0001-6547-1629)
- Zhenming Li (ORCID: https://orcid.org/0000-0002-6752-6264)
- Haibing Li (ORCID: https://orcid.org/0000-0001-5083-8133)
- Xinming Zhang (ORCID: https://orcid.org/0000-0002-6713-1430)
- Jin Duan (ORCID: https://orcid.org/0009-0003-1216-9241)
- Zhongbo Zhang (ORCID: https://orcid.org/0000-0001-5294-444X)
- Bing Zheng
- Chao Dong (ORCID: https://orcid.org/0000-0001-6335-0508)
Institutions
- Changchun University of Science and Technology (CN)
- Foshan University (CN)
- Ministry of Natural Resources (CN)
- Guangzhou Marine Geological Survey (CN)
- Southern Marine Science and Engineering Guangdong Laboratory (Guangzhou) (CN)
- Southern Marine Science and Engineering Guangdong Laboratory (Zhuhai) (CN)
Publication Details
- Journal
- Journal of Marine Science and Engineering
- Published
- 2024-12-12
- DOI
- https://doi.org/10.3390/jmse12122291
- Citations
- 10
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 2.49