UNIR-Net: A Novel Approach for Restoring Underwater Images with Non-Uniform Illumination Using Synthetic Data

Abstract

Restoring underwater images affected by non-uniform illumination (NUI) is essential to improve visual quality and usability in marine applications. Conventional methods often fall short in handling complex illumination patterns, while learning-based approaches face challenges due to the lack of targeted datasets. To address these limitations, the Underwater Non-uniform Illumination Restoration Network (UNIR-Net) is proposed. UNIR-Net integrates multiple components, including illumination enhancement, attention mechanisms, visual refinement, and contrast correction, to effectively restore underwater images affected by NUI. In addition, the Paired Underwater Non-uniform Illumination (PUNI) dataset is introduced, specifically designed for training and evaluating models under NUI conditions. Experimental results on PUNI and the large-scale real-world Non-Uniform Illumination Dataset (NUID) show that UNIR-Net achieves superior performance in both quantitative metrics and visual outcomes. UNIR-Net also improves downstream tasks such as underwater semantic segmentation, highlighting its practical relevance. The code of this method is available at https://github.com/xingyumex/UNIR-Net

This is the preprint version of an article that has been accepted for publication in Image and Vision Computing

References in corpus (5)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• R2RNet: Low-light Image Enhancement via Real-low to Real-normal Network
• Adaptive deep learning framework for robust unsupervised underwater image enhancement
• A Comprehensive Survey on Underwater Image Enhancement Based on Deep Learning
• ALEN: A Dual-Approach for Uniform and Non-Uniform Low-Light Image Enhancement