DPF-Net: Physical Imaging Model Embedded Data-Driven Underwater Image Enhancement

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Mei, Han, Li, Kunqian, Liu, Shuaixin, Ma, Chengzhi, Jiang, Qianli
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917958848086016
author Mei, Han
Li, Kunqian
Liu, Shuaixin
Ma, Chengzhi
Jiang, Qianli
author_facet Mei, Han
Li, Kunqian
Liu, Shuaixin
Ma, Chengzhi
Jiang, Qianli
contents Due to the complex interplay of light absorption and scattering in the underwater environment, underwater images experience significant degradation. This research presents a two-stage underwater image enhancement network called the Data-Driven and Physical Parameters Fusion Network (DPF-Net), which harnesses the robustness of physical imaging models alongside the generality and efficiency of data-driven methods. We first train a physical parameter estimate module using synthetic datasets to guarantee the trustworthiness of the physical parameters, rather than solely learning the fitting relationship between raw and reference images by the application of the imaging equation, as is common in prior studies. This module is subsequently trained in conjunction with an enhancement network, where the estimated physical parameters are integrated into a data-driven model within the embedding space. To maintain the uniformity of the restoration process amid underwater imaging degradation, we propose a physics-based degradation consistency loss. Additionally, we suggest an innovative weak reference loss term utilizing the entire dataset, which alleviates our model's reliance on the quality of individual reference images. Our proposed DPF-Net demonstrates superior performance compared to other benchmark methods across multiple test sets, achieving state-of-the-art results. The source code and pre-trained models are available on the project home page: https://github.com/OUCVisionGroup/DPF-Net.
format Preprint
id arxiv_https___arxiv_org_abs_2503_12470
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DPF-Net: Physical Imaging Model Embedded Data-Driven Underwater Image Enhancement
Mei, Han
Li, Kunqian
Liu, Shuaixin
Ma, Chengzhi
Jiang, Qianli
Computer Vision and Pattern Recognition
Image and Video Processing
Due to the complex interplay of light absorption and scattering in the underwater environment, underwater images experience significant degradation. This research presents a two-stage underwater image enhancement network called the Data-Driven and Physical Parameters Fusion Network (DPF-Net), which harnesses the robustness of physical imaging models alongside the generality and efficiency of data-driven methods. We first train a physical parameter estimate module using synthetic datasets to guarantee the trustworthiness of the physical parameters, rather than solely learning the fitting relationship between raw and reference images by the application of the imaging equation, as is common in prior studies. This module is subsequently trained in conjunction with an enhancement network, where the estimated physical parameters are integrated into a data-driven model within the embedding space. To maintain the uniformity of the restoration process amid underwater imaging degradation, we propose a physics-based degradation consistency loss. Additionally, we suggest an innovative weak reference loss term utilizing the entire dataset, which alleviates our model's reliance on the quality of individual reference images. Our proposed DPF-Net demonstrates superior performance compared to other benchmark methods across multiple test sets, achieving state-of-the-art results. The source code and pre-trained models are available on the project home page: https://github.com/OUCVisionGroup/DPF-Net.
title DPF-Net: Physical Imaging Model Embedded Data-Driven Underwater Image Enhancement
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2503.12470