Weiming Zhuang
Weiming Zhuang
Home
Publications
Posts
Patents
object detection
A Simple Background Augmentation Method for Object Detection with Diffusion Model
We propose a simple yet effective data augmentation approach by leveraging advancements in generative models, specifically text-to-image synthesis technologies like Stable Diffusion. Our method focuses on generating variations of labeled real images, utilizing generative object and background augmentation via inpainting to augment existing training data without the need for additional annotations. We find that background augmentation, in particular, significantly improves the models' robustness and generalization capabilities. We also investigate how to adjust the prompt and mask to ensure the generated content comply with the existing annotations.
Yuhang Li
,
Xin Dong
,
Chen Chen
,
Weiming Zhuang
,
Lingjuan Lyu
PDF
Cite
Cite
×