Wildfire Smoke Detection Enhanced by Image Augmentation with StyleGAN2-ADA for YOLOv8 and RT-DETR Models
Gi-Tae Park, Yangwon Lee
Abstract
Wildfires pose significant environmental and societal threats, necessitating improved early detection methods. This study investigates the effectiveness of integrating real-time object detection deep learning models (YOLOv8 and RT-DETR) with advanced data augmentation techniques, including StyleGAN2-ADA, for wildfire smoke detection. We evaluated model performance on datasets enhanced with fundamental transformations and synthetic images, focusing on detection accuracy. YOLOv8X demonstrated superior overall performance with [email protected] of 0.962 and [email protected] of 0.900, while RT-DETR-X excelled in small object detection with a 0.983 detection rate. Data augmentation, particularly StyleGAN2-ADA, significantly enhanced model performance across various metrics. Our approach reduced average detection times to 1.52 min for YOLOv8X and 2.40 min for RT-DETR-X, outperforming previous methods. The models demonstrated robust performance under challenging conditions, like fog and camera noise, providing reassurance of their effectiveness. While false positives remain a challenge, these advancements contribute significantly to early wildfire smoke detection capabilities, potentially mitigating wildfire impacts through faster response times. This research establishes a foundation for more effective wildfire management strategies and underscores the potential of deep learning applications in environmental monitoring.