Enhancing sonar image segmentation with random fusion in a diffusion model framework
摘要
Sonar images, characterized by low resolution and high noise, present significant challenges for effective segmentation. Existing networks struggle to suppress the inherent noise in these images. We propose SonarRanDiff, a novel segmentation network tailored for sonar images. For the first time, we introduce a diffusion model to sonar image segmentation, incorporating a random fusion module, RanFusion, which boosts convergence speed and improves accuracy by effectively integrating original image and mask information. Comprehensive experiments show that SonarRanDiff surpasses state-of-the-art methods by at least 4.5% on our sonar dataset, underscoring its superior ability to address the unique challenges of sonar imaging. The code is available at: https://github.com/mzh529/SonarRanDiff.git.