PDF(12133 KB)
Fire and smoke detection algorithm based on improved YOLOv8
Li Deng, Jin Zhou, Quanyi Liu
Journal of Tsinghua University(Science and Technology) ›› 2025, Vol. 65 ›› Issue (4) : 681-689.
PDF(12133 KB)
PDF(12133 KB)
Fire and smoke detection algorithm based on improved YOLOv8
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.
fire and smoke / improved YOLOv8 / SlimNeck / slicing-aided hyper inference
| 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 |
|
| 7 |
赵媛媛, 朱军, 谢亚坤, 等. 改进Yolo-v3的视频图像火焰实时检测算法[J]. 武汉大学学报(信息科学版), 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 |
|
| 10 |
|
| 11 |
王云艳, 罗帅, 王子健. 基于改进MobileNetV3的遥感目标检测[J]. 陕西科技大学学报, 2022, 40 (3): 164- 171.
|
| 12 |
袁硕, 刘玉敏, 安志伟, 等. 基于改进ShuffleNetV2网络的岩石图像识别[J]. 吉林大学学报(信息科学版), 2023, 41 (3): 450- 458.
|
| 13 |
|
| 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 |
|
| 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.
|
/
| 〈 |
|
〉 |