Underwater images often suffer from degradations, including color distortion, reduced visibility, and low contrast due to light absorption and scatter in water. Numerous enhancement techniques have been proposed to improve visual quality and address these challenges. However, no single method consistently performs best across all underwater scenes. This work introduces a novel deep learning framework for the automatic selection of the most suitable enhancement technique for underwater images. A novel fused objective metric, combining the Underwater Color Image Quality Evaluation (UCIQE), Underwater Image Quality Measure (UIQM), and Underwater Image Fidelity (UIF) metrics is introduced to assess image quality effectively. The metric is then utilized to train a Shifted Window (Swin) transformer model, which predicts the best enhancement method for each image. This approach advances automatic underwater image enhancement by addressing varying image conditions with a data-driven, adaptive process. Experimental results show that the proposed model achieves an F1 score of 87.88% in selecting the optimal enhancement technique, effectively determining the best enhancement based on the characteristics of the input image.
Vrochidis et al. (Tue,) studied this question.