Underwater Image Enhancement and Small Object Detection Method Based on RBE-CycleGAN and MSFDC-Net
Zongren Li, Chundong Xu, Wenjun Hui, Rui Chen, Xiaofang KongUnderwater object detection plays a vital role in marine exploration and resource exploitation. However, complex underwater environment leads to severe color deviation, blurring, and information loss of small targets, which greatly restrict detection performance. To address these problems, this paper integrates the Channel Attention and Spatial Attention Block (CASAB) attention mechanism into residual blocks based on generative adversarial networks to correct color distortion and improve the clarity of degraded underwater images. For underwater small object detection, MobileNetV2 is selected as the backbone network within the Faster R-CNN framework, and a multi-scale feature fusion strategy is adopted to reduce feature loss caused by repeated downsampling. In the detection head, coordinate attention and parallel dilated convolution are further integrated to suppress background noise and expand the receptive field of feature extraction. Experimental results on the Underwater Robot Professional Contest (URPC) dataset demonstrate that the proposed method yields gains of 10.06%, 9.43%, and 12.29% in three evaluation metrics: Underwater Image Quality Measure (UIQM), Underwater Colour Image Quality Evaluation (UCIQE) and Natural Image Quality Evaluator (NIQE), together with 7.81% in Mean Average Precision (mAP) and an 8.57% increase in Mean Recall (mRecall). These results demonstrate the effectiveness of all improvements.