본문 바로가기
  • Home

GAN-based Image Data Augmentation for Improving Object Detection Performance in Industrial Environments

  • Journal of Software Assessment and Valuation
  • Abbr : JSAV
  • 2022, 18(2), pp.247-259
  • DOI : 10.29056/jsav.2022.12.25
  • Publisher : Korea Software Assessment and Valuation Society
  • Research Area : Engineering > Computer Science
  • Received : November 30, 2022
  • Accepted : December 20, 2022
  • Published : December 31, 2022

Moon-Ki Back 1 Kye Kyung Kim 1

1한국전자통신연구원

Accredited

ABSTRACT

Object detection is one of the important industrial safety technologies that can automatically provide a worker with alerts to avoid unexpected near misses. However, deep learning-based object detection models require large amounts of training data to achieve higher performance, and data collection and labeling work is laborious and requires human resources. To address these limitations, we propose a GAN-based data augmentation that can supplement the original dataset with more diverse examples. In addition, we present a transformer-based generator network to improve the fidelity of generated data and evaluate the existing object detection model(YOLOv5) trained under different augmentation settings for a comparison study. The evaluation results show that the classification ability of the model trained with 20% augmented data has improved by 0.9% without localization performance losses.

Citation status

* References for papers published after 2022 are currently being built.