A Wavelet-based Dual-stream Network for Underwater Image Enhancement
- URL: http://arxiv.org/abs/2202.08758v1
- Date: Thu, 17 Feb 2022 16:57:25 GMT
- Title: A Wavelet-based Dual-stream Network for Underwater Image Enhancement
- Authors: Ziyin Ma and Changjae Oh
- Abstract summary: We present a wavelet-based dual-stream network that addresses color cast and blurry details in underwater images.
We handle these artifacts separately by decomposing an input image into multiple frequency bands using discrete wavelet transform.
We validate the proposed method on both real-world and synthetic underwater datasets and show the effectiveness of our model in color correction and blur removal with low computational complexity.
- Score: 11.178274779143209
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: We present a wavelet-based dual-stream network that addresses color cast and
blurry details in underwater images. We handle these artifacts separately by
decomposing an input image into multiple frequency bands using discrete wavelet
transform, which generates the downsampled structure image and detail images.
These sub-band images are used as input to our dual-stream network that
incorporates two sub-networks: the multi-color space fusion network and the
detail enhancement network. The multi-color space fusion network takes the
decomposed structure image as input and estimates the color corrected output by
employing the feature representations from diverse color spaces of the input.
The detail enhancement network addresses the blurriness of the original
underwater image by improving the image details from high-frequency sub-bands.
We validate the proposed method on both real-world and synthetic underwater
datasets and show the effectiveness of our model in color correction and blur
removal with low computational complexity.
Related papers
- WTDUN: Wavelet Tree-Structured Sampling and Deep Unfolding Network for Image Compressed Sensing [51.94493817128006]
We propose a novel wavelet-domain deep unfolding framework named WTDUN, which operates directly on the multi-scale wavelet subbands.
Our method utilizes the intrinsic sparsity and multi-scale structure of wavelet coefficients to achieve a tree-structured sampling and reconstruction.
arXiv Detail & Related papers (2024-11-25T12:31:03Z) - FDCE-Net: Underwater Image Enhancement with Embedding Frequency and Dual Color Encoder [49.79611204954311]
Underwater images often suffer from various issues such as low brightness, color shift, blurred details, and noise due to absorption light and scattering caused by water and suspended particles.
Previous underwater image enhancement (UIE) methods have primarily focused on spatial domain enhancement, neglecting the frequency domain information inherent in the images.
arXiv Detail & Related papers (2024-04-27T15:16:34Z) - RSFDM-Net: Real-time Spatial and Frequency Domains Modulation Network
for Underwater Image Enhancement [5.3240763486073055]
We propose a Real-time Spatial and Frequency Domains Modulation Network (RSFDM-Net) for the efficient enhancement of colors and details in underwater images.
Our proposed conditional network is designed with Adaptive Fourier Gating Mechanism (AFGM) and Multiscale Conal Attention Module (MCAM)
To more precisely correct the color cast and low saturation of the image, we introduce a Three-branch Feature Extraction (TFE) block in the primary net.
arXiv Detail & Related papers (2023-02-23T17:27:05Z) - Dif-Fusion: Towards High Color Fidelity in Infrared and Visible Image
Fusion with Diffusion Models [54.952979335638204]
We propose a novel method with diffusion models, termed as Dif-Fusion, to generate the distribution of the multi-channel input data.
Our method is more effective than other state-of-the-art image fusion methods, especially in color fidelity.
arXiv Detail & Related papers (2023-01-19T13:37:19Z) - U-shape Transformer for Underwater Image Enhancement [0.0]
In this work, we constructed a large-scale underwater image dataset including 5004 image pairs.
We reported an U-shape Transformer network where the transformer model is for the first time introduced to the UIE task.
In order to further improve the contrast and saturation, a novel loss function combining RGB, LAB and LCH color spaces is designed.
arXiv Detail & Related papers (2021-11-23T13:15:56Z) - WaveFill: A Wavelet-based Generation Network for Image Inpainting [57.012173791320855]
WaveFill is a wavelet-based inpainting network that decomposes images into multiple frequency bands.
WaveFill decomposes images by using discrete wavelet transform (DWT) that preserves spatial information naturally.
It applies L1 reconstruction loss to the low-frequency bands and adversarial loss to high-frequency bands, hence effectively mitigate inter-frequency conflicts.
arXiv Detail & Related papers (2021-07-23T04:44:40Z) - Wavelength-based Attributed Deep Neural Network for Underwater Image
Restoration [9.378355457555319]
This paper shows that attributing the right receptive field size (context) based on the traversing range of the color channel may lead to a substantial performance gain.
As a second novelty, we have incorporated an attentive skip mechanism to adaptively refine the learned multi-contextual features.
The proposed framework, called Deep WaveNet, is optimized using the traditional pixel-wise and feature-based cost functions.
arXiv Detail & Related papers (2021-06-15T06:47:51Z) - Underwater Image Enhancement via Medium Transmission-Guided Multi-Color
Space Embedding [88.46682991985907]
We present an underwater image enhancement network via medium transmission-guided multi-color space embedding, called Ucolor.
Our network can effectively improve the visual quality of underwater images by exploiting multiple color spaces embedding.
arXiv Detail & Related papers (2021-04-27T07:35:30Z) - Wavelet-Based Dual-Branch Network for Image Demoireing [148.91145614517015]
We design a wavelet-based dual-branch network (WDNet) with a spatial attention mechanism for image demoireing.
Our network removes moire patterns in the wavelet domain to separate the frequencies of moire patterns from the image content.
Experiments demonstrate the effectiveness of our method, and we further show that WDNet generalizes to removing moire artifacts on non-screen images.
arXiv Detail & Related papers (2020-07-14T16:44:30Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.