Article
Version 1
Preserved in Portico This version is not peer-reviewed
Deep Dynamic Weights for Underwater Image Restoration
Version 1
: Received: 18 June 2024 / Approved: 19 June 2024 / Online: 19 June 2024 (05:53:43 CEST)
A peer-reviewed article of this Preprint also exists.
Awan, H.S.A.; Mahmood, M.T. Deep Dynamic Weights for Underwater Image Restoration. J. Mar. Sci. Eng. 2024, 12, 1208. Awan, H.S.A.; Mahmood, M.T. Deep Dynamic Weights for Underwater Image Restoration. J. Mar. Sci. Eng. 2024, 12, 1208.
Abstract
Underwater imaging presents unique challenges, notably color distortions and reduced contrast due to light attenuation and scattering. Most underwater image enhancement methods, first use linear transformations for color compensation and then enhance the image. We observed that linear transformation for color compensation is not suitable for certain images. For such images, a non-linear mapping is a better choice. Our paper introduces a unique underwater image restoration approach leveraging a streamlined convolutional neural network (CNN) for dynamic weight learning for linear and non-linear mapping. In the first phase, a classifier is applied that classifies the input images into Type-I or Type-II. In the second phase, depending on the classification, the Deep Line Model (DLM) for Type-I images or the Deep Curve Model (DCM) for Type-II images. For mapping an input image to an output image, DML creatively combines color compensation and contrast adjustment in a single step and uses deep lines for transformation, whereas DCM employs higher-order curves. Both models utilize lightweight neural networks that learn per-pixel dynamic weights based on the input image’s characteristics. Comprehensive evaluations on benchmark datasets using metrics like peak signal-to-noise ratio(PSNR) and root mean squares error (RMSE) affirm our method’s effectiveness in accurately restoring underwater images, outperforming existing techniques.
Keywords
underwater images; underwater image restoration; underwater image enhancement; color restoration; lightweight network; deep learning
Subject
Computer Science and Mathematics, Artificial Intelligence and Machine Learning
Copyright: This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Comments (0)
We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.
Leave a public commentSend a private comment to the author(s)
* All users must log in before leaving a comment