Multi-feature Learning Adaptive Network for Underwater Image Enhancement

Abstract

Underwater image enhancement faces variety of challenges owing to the diversity of underwater scenes (viewed as water types) and the rich multi-frequency information. To deal with these challenges, this paper proposes a multi-feature learning adaptive underwater image enhancement network comprising an adaptive module and a dual-layer synchronous enhancement network. First, we design an adaptive module which enables the determination of water type inside the model and eliminates the negative effect of water type diversity by building water type related features. Then, the model learns high-frequency and low-frequency features through a dual-layer synchronous enhancement network to extract more comprehensive information. Finally, the outputs of the dual-layer network are merged to obtain more realistic underwater enhanced images. Numerous experiments have shown that the proposed method outperforms the comparison method for visual perception and assessment metrics.