OVMamba: omnidirectional vision state space model for underwater images enhancement
摘要
Underwater image enhancement is a crucial technique for improving the quality of underwater photographs, widely applied in fields such as underwater target tracking, robotics, and biological exploration. The unique characteristics of underwater environments often lead to issues like reduced contrast, color distortion, loss of detail, and blurred textures. Therefore, our proposed OVMamba method leverages a global omnidirectional state space to enhance the capture of global information, improving image quality by modeling relationships between pixels and channels. To tackle inconsistent quality, we introduce a multi-head channel attention mechanism, focusing on different subspaces and channels, particularly those with significant quality degradation, to enhance contrast. Additionally, we perform feature extraction independently within RGB color channels and adaptively adjust channel weights to restore natural colors and enhance visual perception. Experimental results demonstrate the effectiveness of our method across multiple datasets.