<p>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.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

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

  • Yaoming Zhuang,
  • Longyu Ma,
  • Jiaming Liu,
  • Yonghao Xian,
  • Baoquan Chen,
  • Li Li,
  • Chengdong Wu,
  • Wei Cui,
  • Zhanlin Liu

摘要

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.