Adversarial enhancement generation method for side-scan sonar images based on DDPM–YOLO

作者：Chengyang Peng, Shaohua Jin, Hanyao Liu, Wei Zhang, Haixing Xia · 发表于：Marine Geodesy · 年份：2024 · DOI：10.1080/01490419.2024.2393190 · 被引用次数：4 · 研究领域：Advanced Image Processing Techniques、Image and Signal Denoising Methods、Generative Adversarial Networks and Image Synthesis

As marine exploration advances, efficient and accurate seabed target identification becomes increasingly critical. However, traditional methods and current technologies face challenges such as scarce samples and complex imaging conditions when dealing with side-scan sonar images. Given the scarcity of sample augmentation methods for side-scan sonar, this paper iteratively trains the Denoising Diffusion Probabilistic Models(DDPM), integrating the DDPM diffusion model and the downstream You Only Look Once(YOLO) retrieval task into a mutually reinforcing framework, proposing an adversarial enhancement generation method based on the DDPM and YOLO detection models. Experiments demonstrate that the DDPM model generated through this adversarial enhancement generation method can improve the accuracy of downstream YOLO target detection tasks by 7%. The images generated by this model also perform optimally on the Fréchet Inception Distance(FID), Maximum Mean Discrepancy(MMD), and Learned Perceptual Image Patch Similarity(LPIPS) metrics, thereby proving that our method can enhance the quality of generated images from the side-scan sonar diffusion model and offer a new avenue for improving the construction of underwater target detection models.