Underwater Image Enhancement Network MIE-Net

Due to the complexity of underwater environments and the diversity of image degradation, underwater image enhancement (UIE) is an extremely challenging task. Existing single-stage networks are difficult to simultaneously solve multiple degradation problems. Therefore, this article proposes a two-stage deep learning framework called Multi-Information-Excitation Network (MIE-Net), which uses cascaded contrastive learning to guide the training of each stage. In the first stage, raw images and reference images are used as negative samples and positive samples, respectively, to establish a contrastive loss for constraining the network training, ensuring that the color correction result is better than the input. In the second stage, the output of the first stage is used as negative samples to ensure that the final enhanced results of the second mist removal stage are better than the intermediate results. Experiments show that  MIE-Net outperforms existing methods on multiple benchmark datasets. The ablation study validated the effectiveness of each key component.