Object Detection with Dataset Augmentation for Fire Images Based on GAN

Objection detection is the task to find and classify objects in images. Many object detection models based on a deep learning algorithm have been proposed. Deep learning algorithms require the models to be trained with affluent images with accurate annotations. However, in the case of fire detection...

Full description

Saved in:
Bibliographic Details
Published in2022 13th International Conference on Information and Communication Technology Convergence (ICTC) pp. 2118 - 2123
Main Authors Lee, Hyungtak, Kang, Seongju, Chung, Kwangsue
Format Conference Proceeding
LanguageEnglish
Published IEEE 19.10.2022
Subjects
Annotations
Classification algorithms
Dataset Augmentation
Deep learning
Fires
Generative Adversarial Network
Generative adversarial networks
Image synthesis
Image-to-Image Translation
Object detection
Online AccessGet full text

Cover

More Information
Summary:Objection detection is the task to find and classify objects in images. Many object detection models based on a deep learning algorithm have been proposed. Deep learning algorithms require the models to be trained with affluent images with accurate annotations. However, in the case of fire detection, neither enough datasets to train the detection model nor correct and sufficient annotations exist. In this paper, we propose a GAN-based model to generate fire images with bounding boxes to enhance the performance of the fire detection model. Through the experiments, we demonstrated that the model can inject flame images into the clean images within specified areas and the generated images are enough to augment the fire detection dataset so that the model's object detection performance can be improved.
ISSN:2162-1241
DOI:10.1109/ICTC55196.2022.9952972