MSCC-RetNet: a multi-scale color corrected retinex network for underwater image enhancement
摘要
High-quality underwater images are crucial for applications such as seafloor exploration, resource monitoring, and object detection. However, underwater imaging faces significant challenges, including poor visibility, low contrast, and severe color distortion. To address these issues, we propose a novel deep learning framework for Underwater Image Enhancement (UIE), incorporating color correction, cross-attention mechanism, and visibility enhancement. The proposed Multi-Scale Color Corrected Retinex Network (MSCC-RetNet) consists of three key modules: a multi-scale color correction module, an information complementarity module, and a multi-scale feature fusion module. The multi-scale color correction module integrates the Inception (Inc) and Squeeze-and-Excitation (SE) modules to enhance feature extraction. The multi-scale feature fusion module combines Convolutional Neural Networks (CNNs) with Transformers to improve representation learning. Experimental results indicate that the multi-scale color correction module effectively mitigates color distortions, while the cross-attention mechanism in the Information Complementarity module selectively emphasizes important regions. Additionally, the multi-scale fusion module enables end-to-end processing of illumination variations, significantly enhancing image visibility. Comprehensive evaluations on five benchmark underwater image datasets confirm that MSCC-RetNet outperforms four traditional approaches and five deep learning-based methods in key image quality assessment metrics.