An Improved Lightweight YOLOv8 Network for Early Small Flame Target Detection

Abstract

The efficacy of early fire detection hinges on its swift response and precision, which allows for the issuance of timely alerts in the nascent stages of a fire, thereby minimizing losses and injuries. To enhance the precision and swiftness of identifying minute early flame targets, as well as the ease of deployment at the edge end, an optimized early flame target detection algorithm for YOLOv8 is proposed. The original feature fusion module, an FPN (feature pyramid network) of YOLOv8n, has been enhanced to become the BiFPN (bidirectional feature pyramid network) module. This modification enables the network to more efficiently and rapidly perform multi-scale fusion, thereby enhancing its capacity for integrating features across different scales. Secondly, the efficient multi-scale attention (EMA) mechanism is introduced to ensure the effective retention of information on each channel and reduce the computational overhead, thereby improving the model’s detection accuracy while reducing the number of model parameters. Subsequently, the NWD (normalized Wasserstein distance) loss function is employed as the bounding box loss function, which enhances the model’s regression performance and robustness. The experimental results demonstrate that the size of the enhanced model is 4.8 M, a reduction of 22.5% compared to the original YOLOv8n. Additionally, the mAP0.5 metric exhibits a 2.7% improvement over the original YOLOv8n, indicating a more robust detection capability and a more compact model size. This makes it an ideal candidate for deployment in edge devices.