Optimized Flame Detection Using a YOLOv5 Neural Network

With the rising prevalence of smart homes, there's an increasing demand for comprehensive automation solutions to mitigate fire risks, especially when homeowners are absent or in homes with elderly residents. This study focuses on enhancing flame detection techniques, particularly for smaller, hard-to-detect fire sources. A customized Yolov5 neural network was developed to improve detection precision. This enhancement was facilitated using a dataset of over 2,000 images from lab camera observations and online platforms. These images were then manually labeled and clustered for anchor dimensions using the K-means++ algorithm. The CIOU and diou _ nms techniques were integrated to optimize the network's performance. Upon training, the refined model was tested on the Jet Nano, revealing its high efficacy in flame detection.