arrow
Return

A diffusion model-based image generation framework for underwater object detection

delete2025-12-29
delete0
PRE
AI
庄
庄曜铭 (Yaoming Zhuang) *
L
Longyu Ma
L
Liu, Jiaming
Y
Yonghao Xian
B
Baoquan Chen
L
Li Li
C
Chengdong Wu
W
Wei Cui
Z
Zhanlin Liu
DOI:10.1038/s44172-025-00579-zdelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Underwater object detection plays a crucial role in applications such as marine ecological monitoring and underwater rescue operations. However, challenges such as limited underwater data availability and low scene diversity hinder detection accuracy. In this paper, we propose the Underwater Layout-Guided Diffusion Framework (ULGF), a diffusion model-based framework designed to augment underwater detection datasets. Unlike conventional methods that generate underwater images by integrating in-air information, ULGF operates exclusively on a small set of underwater images and their corresponding labels, requiring no external data. We have publicly released the ULGF source code and the generated dataset for further research. Our approach enables the generation of high-fidelity, diverse, and theoretically infinite underwater images, substantially enhancing object detection performance in real-world underwater scenarios. Furthermore, we evaluate the quality of the generated underwater images, demonstrating that ULGF produces images with a smaller domain gap.
Keywords:
Underwater object detection
Diffusion model
Dataset augmentation
Image generation
Domain gap

Journal

C
Communications Engineering
IF:
0
Papers:
176
Citations:
1

Organization

A
a*star - institute for infocomm research (i2r)
Scholars:
869
Papers: 880
Citations: 1
N
northeastern university - china
Scholars:
3.2W
Papers: 2.7W
Citations: 37
A
agency for science technology & research (a*star)
Scholars:
2.2W
Papers: 1.9W
Citations: 57
researcher View more organizations
Cited Papers

Cited Papers

U-Net: Convolutional Networks for Biomedical Image Segmentation
err2015-11-18
err0
PREAI
errOlaf Ronneberger; Philipp Fischer; Thomas Brox
errShare
errSave
Unveiling the underwater world: CLIP perception model-guided underwater image enhancement
err2025-06-01
err0
PREAI
errCao, Jiangzhong; Zeng, Zekai; Zhang, Xu; Fan, Chunling; Jiang, Gangyi; Lin, Weisi
errShare
errSave
High-Resolution Image Synthesis with Latent Diffusion Models
err2022-06-01
err0
errOAAI
errRobin Rombach; Andreas Blattmann; Dominik Lorenz; Patrick Esser; Bjorn Ommer
errShare
errSave
errShare
errSave
An Underwater Image Enhancement Benchmark Dataset and Beyond
err2020-01-01
err1.1K
errOAAI
errLi, Chongyi; Guo, Chunle; Ren, Wenqi; Cong, Runmin; Hou, Junhui; Kwong, Sam; Tao, Dacheng
errShare
errSave
Underwater Image Enhancement via Minimal Color Loss and Locally Adaptive Contrast Enhancement
err2022-01-01
err315
PREAI
errZhang, Weidong; Zhuang, Peixian; Sun, Hai-Han; Li, Guohou; Kwong, Sam; Li, Chongyi
errShare
errSave
researcher View more