An Efficient Fire Detection Model Based on Optimized YOLOv5 Leveraging Custom Data Approaches
Loading...
Date
2024
Authors
Journal Title
Journal ISSN
Volume Title
Publisher
UMT, Lahore
Abstract
Fires pose significant threats across various contexts, emphasizing the critical need for accurate and efficient fire detection systems. Deep learning methodologies, particularly YOLO (You Only Look Once), have demonstrated promising capabilities in real-time fire detection applications. In this study, we present an advanced iteration of the YOLOv5s architecture tailored specifically for fire detection tasks. Our approach integrates state-of-the-art techniques to enhance the detection accuracy and efficiency of small-scale fire instances. We introduce novel components, including coordinate attention mechanisms and refined loss functions, to improve the YOLOv5s model's performance to capture intricate fire patterns and accurately identify fire targets.