A Dual-Branch Collaborative Framework for Joint Optimization of Underwater Image Enhancement and Object Detection

• URL: http://arxiv.org/abs/2606.15857v1
• Date: Sun, 14 Jun 2026 15:25:31 GMT
• Title: A Dual-Branch Collaborative Framework for Joint Optimization of Underwater Image Enhancement and Object Detection
• Abstract summary: This paper proposes an efficient dual-branch underwater image enhancement framework for object detection.<n>The detail enhancement branch improves brightness and local contrast to recover texture details in dark regions.<n>The color restoration branch uses adaptive compensation to reduce color distortion and improve color gradation.
• Score: 13.283059746689377
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Due to wavelength dependent light absorption and scattering, underwater images usually suffer from color distortion and blurred details, which limits underwater object detection performance. Existing underwater image enhancement methods mainly focus on visual quality improvement, while it is still difficult to balance enhancement quality, processing efficiency, and downstream detection performance. Therefore, this paper proposes an efficient dual-branch underwater image enhancement framework for object detection. The detail enhancement branch improves brightness and local contrast to recover texture details in dark regions. The color restoration branch uses adaptive compensation to reduce color distortion and improve color gradation. By combining the complementary outputs of the two branches, the proposed framework provides clearer and more informative images for object detection. On the UIEB and EUVP datasets, the proposed method achieves UIQM scores of 2.249 and 2.576. When applied to the YOLOv8 detection task on the URPC dataset, the proposed method improves mAP50 by 2.1\% compared with the baseline. Extensive experiments show that our method improves object detection in complex underwater scenes, while balancing enhancement quality and processing efficiency.

Related papers

• Advancing Visual Reliability: Color-Accurate Underwater Image Enhancement for Real-Time Underwater Missions [52.04969451806431]
  We present a real-time underwater image enhancement framework with accurate color restoration.<n>The proposed method achieves state-of-the-art performance across seven evaluation metrics.<n>Our method improves the UCIQE score by 29.7% under diverse environmental conditions.
  arXiv  Detail & Related papers  (2026-03-17T10:51:00Z)
• DACA-Net: A Degradation-Aware Conditional Diffusion Network for Underwater Image Enhancement [16.719513778795367]
  Underwater images typically suffer from severe colour distortions, low visibility, and reduced structural clarity due to complex optical effects such as scattering and absorption.<n>Existing enhancement methods often struggle to adaptively handle diverse degradation conditions and fail to leverage underwater-specific physical priors effectively.<n>We propose a degradation-aware conditional diffusion model to enhance underwater images adaptively and robustly.
  arXiv  Detail & Related papers  (2025-07-30T09:16:07Z)
• Layer Decomposition and Morphological Reconstruction for Task-Oriented Infrared Image Enhancement [4.06181861004177]
  Infrared image helps improve the perception capabilities of autonomous driving in complex weather conditions such as fog, rain, and low light.<n>However, infrared image often suffers from low contrast, especially in non-heat-emitting targets like bicycles.<n>We propose a task-oriented infrared image enhancement method to achieve contrast enhancement without amplifying noise and losing important information.
  arXiv  Detail & Related papers  (2025-06-29T18:10:05Z)
• LUIEO: A Lightweight Model for Integrating Underwater Image Enhancement and Object Detection [10.572090127928698]
  This paper proposes a multi-task learning method that simultaneously enhances underwater images and improves detection accuracy.<n>The integrated model allows for the dynamic adjustment of information communication and sharing between different tasks.<n> Numerical experiments demonstrate that the proposed model achieves satisfactory results in visual performance, object detection accuracy, and detection efficiency.
  arXiv  Detail & Related papers  (2024-12-01T14:01:30Z)
• HUPE: Heuristic Underwater Perceptual Enhancement with Semantic Collaborative Learning [62.264673293638175]
  Existing underwater image enhancement methods primarily focus on improving visual quality while overlooking practical implications.<n>We propose a invertible network for underwater perception enhancement, dubbed H, which enhances visual quality and demonstrates flexibility in handling other downstream tasks.
  arXiv  Detail & Related papers  (2024-11-27T12:37:03Z)
• Dual High-Order Total Variation Model for Underwater Image Restoration [13.789310785350484]
  Underwater image enhancement and restoration (UIER) is one crucial mode to improve the visual quality of underwater images. We propose an effective variational framework based on an extended underwater image formation model (UIFM) In our proposed framework, the weight factors-based color compensation is combined with the color balance to compensate for the attenuated color channels and remove the color cast.
  arXiv  Detail & Related papers  (2024-07-20T13:06:37Z)
• Learning Heavily-Degraded Prior for Underwater Object Detection [59.5084433933765]
  This paper seeks transferable prior knowledge from detector-friendly images. It is based on statistical observations that, the heavily degraded regions of detector-friendly (DFUI) and underwater images have evident feature distribution gaps. Our method with higher speeds and less parameters still performs better than transformer-based detectors.
  arXiv  Detail & Related papers  (2023-08-24T12:32:46Z)
• WaterFlow: Heuristic Normalizing Flow for Underwater Image Enhancement and Beyond [52.27796682972484]
  Existing underwater image enhancement methods mainly focus on image quality improvement, ignoring the effect on practice. We propose a normalizing flow for detection-driven underwater image enhancement, dubbed WaterFlow. Considering the differentiability and interpretability, we incorporate the prior into the data-driven mapping procedure.
  arXiv  Detail & Related papers  (2023-08-02T04:17:35Z)
• PUGAN: Physical Model-Guided Underwater Image Enhancement Using GAN with Dual-Discriminators [120.06891448820447]
  How to obtain clear and visually pleasant images has become a common concern of people. The task of underwater image enhancement (UIE) has also emerged as the times require. In this paper, we propose a physical model-guided GAN model for UIE, referred to as PUGAN. Our PUGAN outperforms state-of-the-art methods in both qualitative and quantitative metrics.
  arXiv  Detail & Related papers  (2023-06-15T07:41:12Z)
• A Correction-Based Dynamic Enhancement Framework towards Underwater Detection [5.103759034134962]
  We propose a lightweight dynamic enhancement algorithm using a contribution dictionary to guide low-level corrections. Experimental results in real underwater object detection tasks show the superiority of our proposed method in both generalization and real-time performance.
  arXiv  Detail & Related papers  (2023-02-06T04:11:52Z)
• Degrade is Upgrade: Learning Degradation for Low-light Image Enhancement [52.49231695707198]
  We investigate the intrinsic degradation and relight the low-light image while refining the details and color in two steps. Inspired by the color image formulation, we first estimate the degradation from low-light inputs to simulate the distortion of environment illumination color, and then refine the content to recover the loss of diffuse illumination color. Our proposed method has surpassed the SOTA by 0.95dB in PSNR on LOL1000 dataset and 3.18% in mAP on ExDark dataset.
  arXiv  Detail & Related papers  (2021-03-19T04:00:27Z)

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.