Research on an Underwater Object Detection Network Based on Dual-Branch Feature Extraction

Abstract

Underwater object detection is challenging in computer vision research due to the complex underwater environment, poor image quality, and varying target scales, making it difficult for existing object detection networks to achieve high accuracy in underwater tasks. To address the issues of limited data and multi-scale targets in underwater detection, we propose a Dual-Branch Underwater Object Detection Network (DB-UODN) based on dual-branch feature extraction. In the feature extraction stage, we design a dual-branch structure by combining the You Only Look Once (YOLO) v7 backbone with the Enhanced Channel and Dilated Block (ECDB). It allows for the extraction and complementation of multi-scale features, which enable the model to learn both global and local information and enhance its perception of multi-scale features in underwater targets. Furthermore, we employ the DSPACSPC structure to replace the SPPCSPC structure in YOLOv7. The DSPACSPC structure utilizes atrous convolutions with different dilation rates to capture contextual information at various scales, compensating for potential information loss caused by pooling operations. Additionally, we utilize a dense connection structure to facilitate feature reuse and enhance the network’s representation and generalization capabilities. Experimental results demonstrate that the proposed DB-UODN outperforms the most commonly used object detection networks in underwater scenarios. On the URPC2020 dataset, the network achieves an average detection accuracy of 87.36%.