Research on an Improved YOLOv8-Based Object Detection Algorithm for Flame and Smoke Detection in Factory Environments

Abstract

Overcoming complex background noise and poor small-target detection in industrial settings, this paper introduces YOLOv8-BBP2, an enhanced YOLOv8 model. To better extract dynamic features, the backbone integrates a BiFormer dual-level routing attention mechanism. Moreover, a learnable Bi-directional Feature Pyramid Network (BiFPN) replaces the standard module, optimizing multi-scale feature integration. A P2 detection head is also added to accurately identify tiny objects, such as early flames and thin smoke. Tested on a custom factory fire dataset, YOLOv8-BBP2 yields 95.231% precision, 94.612% recall, and 89.677% mean average precision (mAP@0.5). These metrics represent respective gains of 3.31%, 4.934%, and 7.451% over the baseline YOLOv8s. Ultimately, with an inference speed of 20 ms per frame, the proposed network ensures highly robust, real-time performance.