Multi-scale fire target detection algorithm using YOLO-fire

Abstract

Accurate and real-time fire detection is crucial for industrial production and daily life. However, the variable form of fire and the significant differences in visual characteristics across its different stages pose great challenges to precise fire prevention and control. To address this issue, a multi-scale fire target detection algorithm using YOLO-fire was proposed by improving the YOLOv8 model. This model introduced new layer structures and attention mechanism, replaced new feature fusion modules and loss functions. By introducing a small-target detection P2 layer, the model’s ability to detect early-stage fires is improved. The coordinate attention mechanism is integrated into the layer structures of multi-scale target detection, enhancing the capture of target location information and channel relationships, thereby focusing more on the target regions. The Neck network structure was optimized by adopting a BiFPN_F strategy for different feature layers, which strengthened the cross-scale representation of fire features and controlled the parameter count of the designed model. The WIoU loss function was employed to optimize the regression process, improving fire source localization accuracy in complex scenarios, enhancing model robustness, and increasing detection precision. Experimental results on fire datasets demonstrated that YOLO-fire could effectively detect multi-scale fire targets in various scenarios. Compared to the baseline model (YOLOv8n), YOLO-fire achieves improvements of 1.37% in accuracy, 1.25% in mAP50-95, and 0.35% in F1-score, while reducing parameters by 3.79%. Furthermore, compared to current mainstream target detection algorithms, YOLO-Fire achieved optimal detection performance while reducing network parameters and computational complexity. This research provided effective technical support for fire safety prevention and control in related fields.

Keywords

fire detection / early-stage fire / feature fusion / attention mechanism / loss function / network structure / YOLOv8

References

[1] VASCONCELOS R N, FRANCA ROCHA W J S, COSTA D P, et al. Fire detection with deep learning: a comprehensive review. Land, 2024, 13(10): 1696.

[2] ZHENG H T, WANG G Y, XIAO D, et al. FTA-DETR: an efficient and precise fire detection framework based on an end-to-end architecture applicable to embedded platforms. Expert Systems with Applications, 2024, 248: 123394.

[3] ZHAO H Y, JIN J, LIU Y, et al. FSDF: a high-performance fire detection framework. Expert Systems with Applications, 2024, 238: 121665.

[4] SUN J P, SUN Y Y, FAN W Q. Mine exogenous fire identification method based on visible light and infrared image. Industry and Mine Automation, 2019, 45(5): 1-5.

[5] FAN W Q, LI X Y, LIU Y, et al. Mine external fire monitoring method using the fusion of visible visual features. Journal of Mining Science and Technology, 2023, 8(4): 529-537.

[6] LI X Y, FAN W Q, LIU Y, et al. Mine exogenous fire monitoring method using the fusion of infrared visual features. Journal of Mining Science and Technology, 2025, 10(1): 116-124.

[7] GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation//2014 IEEE Conference on Computer Vision and Pattern Recognition, June 18-24, 2014, Columbus, Ohio, USA. New York: IEEE, 2014: 580-587.

[8] GIRSHICK R. Fast R-CNN//2015 IEEE International Conference on Computer Vision, December 11-18, 2015, Santiago, Chile. New York: IEEE, 2015: 1440–1448.

[9] REN S Q, HE K M, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6): 1137-1149.

[10] YU L C, LIU J Q. Fire image recognition algorithm based on improved mask R-CNN. Computer Engineering and Applications, 2020, 56(21): 194-198.

[11] ZHANG S L, ZHANG Y N, TIAN C, et al. Study on flame image recognition of chemical industrial park fires based on convolutional neural network. China Safety Science Journal, 2024, 34(1): 179-186.

[12] BARMPOUTIS P, DIMITROPOULOS K, KAZA K, et al. Fire detection from images using faster R-CNN and multidimensional texture analysis//2019 IEEE International Conference on Acoustics, Speech and Signal Processing, May 12-17, 2019. Brighton, KingdomUnited. New York: IEEE, 2019: 8301-8305.

[13] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection//2016 IEEE Conference on Computer Vision and Pattern Recognition, June 27-30, 2016, Las Vegas, NV, USA. New York: IEEE, 2016: 779-788.

[14] REDMON J, FARHADI A. YOLO9000: better, faster, stronger//2017 IEEE Conference on Computer Vision and Pattern Recognition, July 21-26, 2017, Honolulu, HI, USA. New York: IEEE, 2017: 6517-6525.

[15] LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot MultiBox detector//Computer Vision-ECCV 2016. Cham: Springer International Publishing, 2016: 21-37.

[16] GE Z, LIU S, WANG F, et al. YOLOX: Exceeding YOLO series in 2021. (2021)[2024-10-08]. arXiv:2107.08430.

[17] QIAN H M, SHI F, CHEN W, et al. A fire monitoring and alarm system based on channel-wise pruned YOLOv3. Multimedia Tools and Applications, 2022, 81(2): 1833-1851.

[18] ZHAO L, ZHI L Q, ZHAO C, et al. Fire-YOLO: a small target object detection method for fire inspection. Sustainability, 2022, 14(9): 4930.

[19] CHENG S H, MA J Y, ZHANG S J, et al. Smoke detection algorithm combined with improved Gaussian mixture and YOLOv2. Acta Metrologica Sinica, 2019, 40(5): 798-803.

[20] FAN R X, PEI M T. Lightweight forest fire detection based on deep learning//2021 IEEE 31st International Workshop on Machine Learning for Signal Processing, October 25-28, 2021, Gold Coast, Australia. New York: IEEE, 2021: 1-6.

[21] AKHMEDOV F, NASIMOV R, ABDUSALOMOV A. Dehazing algorithm integration with YOLO-v10 for ship fire detection. Fire, 2024, 7(9): 332.

[22] HE N L, ZHANG J S, LIN W S. Research on forest fire image recognition based on deep learning multi-object detection technology. Journal of Nanjing Forestry University (Natural Sciences Edition), 2024, 48(3): 207-218.

[23] HE Y, HU J J, ZENG M, et al. DCGC-YOLO: the efficient dual-channel bottleneck structure YOLO detection algorithm for fire detection. IEEE Access, 2024, 12: 65254-65265.

[24] LIU D, ZHAO X, FAN W Q. A small object detection algorithm for mine environment. Engineering Applications of Artificial Intelligence, 2025, 153: 110936.

[25] CHU Y, WEI Y L, LIANG H G, et al. Lightweight YOLOv8 algorithm for weed target detection. Journal of North University of China (Natural Science Edition), 2025, 46(4): 489-498.

[26] HOU Q B, ZHOU D Q, FENG J S. Coordinate attention for efficient mobile network design//2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition, June 20-25, 2021. Nashville, TN, USA. New York: IEEE, 2021: 13708-13717.

[27] JOCHER G, STOKEN A, CHAURASIA A, et al. ultralytics/yolov5: v6.0—YOLOv5n “Nano” models, Roboflow integration, TensorFlow export, OpenCV DNN support. (2021)[2024-10-08].

[28] JOCHER G, CHAURASIA A, QIU J. YOLO by Ultralytics. (2023, Version 8.0.0)[2024-10-08].

[29] WANG A, CHEN H, LIU L H, et al. YOLOv10: real-time end-to-end object detection. 2024: arXiv: 2405.14458.

[30] LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection//2017 IEEE International Conference on Computer Vision, October 22-29, 2017, Venice, Italy. New York: IEEE, 2017: 2999-3007.

[31] TAN M X, PANG R M, LE Q V. EfficientDet: scalable and efficient object detection//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, June 14-19, 2020. Seattle, WA, USA. New York: IEEE, 2020: 10781-10790.

[32] DUAN K W, BAI S, XIE L X, et al. CenterNet: keypoint triplets for object detection//2019 IEEE/CVF International Conference on Computer Vision, October 27-November 2, 2019. Seoul, Korea. New York: IEEE, 2019: 6568-6577.

[33] ZHAO Y A, LÜ W Y, XU S L, et al. DETRs beat YOLOs on real-time object detection//2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition, June 17-21, 2024, Seattle, WA, USA. New York: IEEE, 2024: 16965-16974.