基于改进YOLOv8的火焰与烟雾检测算法

邓力, 周进, 刘全义

清华大学学报(自然科学版) ›› 2025, Vol. 65 ›› Issue (4) : 681-689.

PDF(12133 KB)
PDF(12133 KB)
清华大学学报(自然科学版) ›› 2025, Vol. 65 ›› Issue (4) : 681-689. DOI: 10.16511/j.cnki.qhdxxb.2024.27.036
森林火灾

基于改进YOLOv8的火焰与烟雾检测算法

作者信息 +

Fire and smoke detection algorithm based on improved YOLOv8

Author information +
文章历史 +

摘要

由于火灾具有快速蔓延的特性和较高的破坏力,实现火灾的早期探测是十分必要的,针对火灾检测算法的研究也尤为重要。该文提出了一种改进的YOLOv8算法,通过集成轻量型模块SlimNeck和切片辅助推理方法SAHI,分别优化了YOLOv8算法的网络结构和推理框架,将火灾数据集目标分类为火焰(fire)、烟雾(smoke)和干扰项(default)。实验结果表明,SlimNeck-YOLOv8算法比相关的先进算法具有更优的火灾检测性能,与YOLOv8模型相比,查全率(recall)增长了2.7%、平均精度(mAP)增长了0.2%,检测速度提高了35 fps,同时也降低了计算负担。在SlimNeck-YOLOv8基础上进一步优化推理框架所得的SlimNeck-YOLOv8+SAHI算法,有效改善了漏检与误检现象。该研究有助于提升火灾检测系统的速度和精度,为火灾预警工作提供了有力的技术支持。

Abstract

Objective: With the rapid and continuous advancement of urbanization at an astonishing pace, fire accidents are happening with increasing frequency globally. A sudden fire outbreak holds a significantly high probability of causing extensive and severe harm to society. Research conducted on image-based fire detection algorithms is highly beneficial and valuable in terms of extracting the detailed morphological features of fires or smoke, aiding in effectively improving the efficiency of fire warnings. Methods: This study presents and introduces an improved version of the YOLOv8 algorithm. Initially, the neck network of the algorithm is strengthened by integrating the SlimNeck lightweight module. Then, the inference framework of the YOLOv8 algorithm is substituted with slicing-aided hyper inference (SAHI) to further enhance the capability of the algorithm to detect small targets. Moreover, fire and smoke are two crucial target categories in fire scenarios. Given the inherent complexity of fire image backgrounds, which frequently contain numerous interferences from nonfire categories, fire dataset targets are classified as fire, smoke, and default. Results: Experimental results clearly indicate that the SlimNeck-YOLOv8 algorithm showcases superior fire detection performance compared with other related advanced algorithms. In contrast to the YOLOv8 algorithm, the recall rate of this algorithm is elevated by 2.7%, mean average precision (mAP) is increased by 0.2%, and detection speed is accelerated by 35 frames/s. Simultaneously, with the developed algorithm, the computational burden is effectively reduced. Conclusions: By integrating SlimNeck and SAHI, respectively, to optimize the network structure and inference framework of the YOLOv8 algorithm, the improved YOLOv8 algorithm is utilized for detecting fire and smoke, which has, to a certain extent, remedied the shortcomings of the YOLOv8 algorithm for this purpose. To effectively verify the performance and effectiveness of the proposed algorithm, the model is not merely trained on the fire dataset but is trained on coco128 dataset under precisely the same training epochs and parameters. This is done with the specific aim of conducting comprehensive tests to accurately evaluate model performance. The improved algorithm proposed in this study has successfully achieved the expected goals of significantly enhancing the mAP, recall, and speed of the YOLOv8 algorithm for detecting fire and smoke and concurrently reducing the rates of missed and false detections. This advancement holds great promise for enhancing the reliability and effectiveness of fire detection systems, providing prior and more accurate warnings to minimize potential losses and damages caused by fires. The combination of innovative techniques and targeted optimizations presented in this research offers valuable insights and practical solutions in the fire safety field and related applications.

关键词

火焰与烟雾 / 改进的YOLOv8 / SlimNeck / 切片辅助超推理

Key words

fire and smoke / improved YOLOv8 / SlimNeck / slicing-aided hyper inference

引用本文

导出引用
邓力, 周进, 刘全义. 基于改进YOLOv8的火焰与烟雾检测算法[J]. 清华大学学报(自然科学版). 2025, 65(4): 681-689 https://doi.org/10.16511/j.cnki.qhdxxb.2024.27.036
Li Deng, Jin Zhou, Quanyi Liu. Fire and smoke detection algorithm based on improved YOLOv8[J]. Journal of Tsinghua University(Science and Technology). 2025, 65(4): 681-689 https://doi.org/10.16511/j.cnki.qhdxxb.2024.27.036
中图分类号: TP212   

参考文献

1
GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]// Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Columbus, USA: IEEE, 2014: 580-587.
2
GIRSHICK R. Fast R-CNN[C]// Proceedings of IEEE International Conference on Computer Vision. Santiago, Chile: IEEE, 2015: 1440-1448.
3
REDMON J, FARHADI A. YOLO9000: Better, faster, stronger[C]// Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Honolulu, USA: IEEE, 2017: 6517-6525.
4
REDMON J, FARHADI A. YOLO9000: Better, faster, stronger[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Honolulu, USA: IEEE, 2017: 6517-6525.
5
WANG C Y, BOCHKOVSKIY A, LIAO H-Y M. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]// Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition. Vancouver, Canada: IEEE, 2023: 7464-7475.
6
DAI Z T . Image flame detection method based on improved YOLOv3[J]. IOP Conference Series: Earth and Environmental Science, 2021, 693, 012012.
7
赵媛媛, 朱军, 谢亚坤, 等. 改进Yolo-v3的视频图像火焰实时检测算法[J]. 武汉大学学报(信息科学版), 2021, 46 (3): 326- 334.
ZHAO Y Y , ZHU J , XIE Y K , et al. A real-time video flame detection algorithm based on improved Yolo-v3[J]. Geomatics and Information Science of Wuhan University, 2021, 46 (3): 326- 334.
8
UDDIN M N, SAKIB M S I, NAWER S, et al. Improved fire detection by YOLOv8 and YOLOv5 to enhance fire safety[C]// Proceedings of the 2023 26th International Conference on Computer and Information Technology (ICCIT). Cox's Bazar, Bangladesh: IEEE, 2023: 1-6.
9
YAR H , KHAN Z A , ULLAH F U M , et al. A modified YOLOv5 architecture for efficient fire detection in smart cities[J]. Expert Systems with Applications, 2023, 231, 120465.
10
QIN Y Y , CAO J T , JI X F . Fire detection method based on depthwise separable convolution and YOLOv3[J]. International Journal of Automation and Computing, 2021, 18 (2): 300- 310.
11
王云艳, 罗帅, 王子健. 基于改进MobileNetV3的遥感目标检测[J]. 陕西科技大学学报, 2022, 40 (3): 164- 171.
WANG Y Y , LUO S , WANG Z J . Remote sensing target detection based on improved MobileNetV3[J]. Journal of Shaanxi University of Science & Technology, 2022, 40 (3): 164- 171.
12
袁硕, 刘玉敏, 安志伟, 等. 基于改进ShuffleNetV2网络的岩石图像识别[J]. 吉林大学学报(信息科学版), 2023, 41 (3): 450- 458.
YUAN S , LIU Y M , AN Z W , et al. Rock image recognition based on improved ShuffleNetV2 network[J]. Journal of Jilin University (Information Science Edition), 2023, 41 (3): 450- 458.
13
LI H , LI J , WEI H , et al. Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles[J]. arXiv preprint arXiv: 2206.02424, 2022,
14
AKYON F C, ALTINUC S O, TEMIZEL A. Slicing aided hyper inference and fine-tuning for small object detection[C]// Proceedings of 2022 IEEE International Conference on Image Processing (ICIP). Bordeaux, France: IEEE, 2022: 966-970.
15
LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Honolulu, USA: IEEE, 2017: 936-944.
16
LIU S, QI L, QIN H F, et al. Path aggregation network for instance segmentation[C]// Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Salt Lake City, USA: IEEE, 2018: 8759-8768.
17
HUANG G, LIU Z, VAN DER MAATEN L, et al. Densely connected convolutional networks[C]// Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Honolulu, USA: IEEE, 2017: 2261-2269.
18
LEE Y, HWANG J W, LEE S, et al. An energy and GPU-computation efficient backbone network for real-time object detection[C]// Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops. Long Beach, USA: IEEE, 2019: 752-760.
19
WANG C Y, LIAO H Y M, WU Y H, et al. CSPNet: A new backbone that can enhance learning capability of CNN[C]// Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops. Seattle, USA: IEEE, 2020: 1571-1580.
20
NEUBECK A, VAN GOOL L. Efficient non-maximum suppression[C]// Proceedings of the 18th International Conference on Pattern Recognition (ICPR'06). Hong Kong, China: IEEE, 2006: 850-855.
21
ABONIA S. YOLOv8-fire-and-smoke-detection[EB/OL]. [2024-03-26]. https://github.com/Abonia1/YOLOv8-Fire-and-Smoke-Detection/tree/main/datasets/fire-8.
22
LIU W, ANGUELOV D, ERHAN D, et al. SSD: Single shot MultiBox detector[C]// Proceedings of 14th European Conference on Computer Vision. Amsterdam, The Netherlands: Springer, 2016: 21-37.
23
CUI L S , MA R , LV P , et al. MDSSD: multi-scale deconvolutional single shot detector for small objects[J]. Science China Information Sciences, 2020, 63 (2): 120113.
24
SELVARAJU R R, COGSWELL M, DAS A, et al. Grad-CAM: Visual explanations from deep networks via gradient-based localization[C]// Proceedings of IEEE International Conference on Computer Vision. Venice, Italy: IEEE, 2017: 618-626.

基金

国家自然科学基金民航联合研究基金(U2033206)
四川省重点实验室项目(MZ2022JB01)
航空科学基金(ASFC-20200046117001)

版权

版权所有,未经授权,不得转载。
PDF(12133 KB)

Accesses

Citation

Detail

段落导航
相关文章

/