Object detection is a critical task in computer vision that can be hampered by limited training data. To overcome data insufficiency, data augmentation has been widely used, traditionally applying geometric transformations or pixel manipulations to existing images. However, these methods lack diversity as they do not introduce new information fundamentally. Text-to-image algorithms, which combine diffusion models with prompts, can generate a more diverse range of images. However, due to potential inaccuracies or redundancies in the prompts, the generated images may deviate from reality. Simultaneously, constructing effective prompts remains a challenge, as it requires carefully selecting relevant keywords while eliminating redundant information. To tackle these limitations, we introduce a novel method called DiffGAN, which merges the diversity capabilities of diffusion models with the reality-enhancing mechanisms of Generative Adversarial Networks (GANs) in an adversarial learning process. DiffGAN utilizes a GAN generator to create text features aligned with the desired image and a GAN discriminator to assess the reality of the images produced by the diffusion model, thereby iteratively refining the GAN generator. Our experiments on the Open Images Dataset V3 and a Diseased Leaf dataset demonstrate that DiffGAN outperforms traditional data augmentation methods, improving object detection metrics (mAP@0.5) by 13.9% and 16.1%, respectively.

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

A Novel Data Synthesis Method by Integration of Diffusion Model and GAN for Object Detection Task

  • Jiarui Xie,
  • Anxin Li,
  • Ryosuke Mizuno,
  • Keisuke Nakamura,
  • Yuusuke Fukushima,
  • Issei Nakamura

摘要

Object detection is a critical task in computer vision that can be hampered by limited training data. To overcome data insufficiency, data augmentation has been widely used, traditionally applying geometric transformations or pixel manipulations to existing images. However, these methods lack diversity as they do not introduce new information fundamentally. Text-to-image algorithms, which combine diffusion models with prompts, can generate a more diverse range of images. However, due to potential inaccuracies or redundancies in the prompts, the generated images may deviate from reality. Simultaneously, constructing effective prompts remains a challenge, as it requires carefully selecting relevant keywords while eliminating redundant information. To tackle these limitations, we introduce a novel method called DiffGAN, which merges the diversity capabilities of diffusion models with the reality-enhancing mechanisms of Generative Adversarial Networks (GANs) in an adversarial learning process. DiffGAN utilizes a GAN generator to create text features aligned with the desired image and a GAN discriminator to assess the reality of the images produced by the diffusion model, thereby iteratively refining the GAN generator. Our experiments on the Open Images Dataset V3 and a Diseased Leaf dataset demonstrate that DiffGAN outperforms traditional data augmentation methods, improving object detection metrics (mAP@0.5) by 13.9% and 16.1%, respectively.