Domain-guided multi-frequency underwater image enhancement network

Abstract

The distribution of underwater images exhibits diverse due to the varied scattering and absorption of light in different water types. However, most existing methods have significant limitations as they cannot distinguish the difference between different water types during enhancement processing, and do not propose clear solutions for the different frequency information. Therefore, the key challenge is to achieve consistency between learned features and water types while preserving multi-frequency information. Thus, we propose a domain-guided multi-frequency underwater image enhancement network (DGMF), which generate high quality images by learning water-type-related features and capturing multi-frequency information. Specifically, we introduce a domain-aware module equipped with a water type classifier, which can distinguish the impacts of different water types, and guide the update of the model towards the specific domain. In addition, we design a multi-frequency mixer that couples Multi-Group Convolution (MGC) and Global Sparse Attention (GSA) to more effectively captures local and global information. Extensive experiments demonstrate that our method outperforms most state-of-the-art methods in both visual perception and evaluation metrics. The code is publicly available at https://github.com/liyoucai699/DGMF.git.