清华大学学报（自然科学版）2025，Vol.65Issue(4)：681-689,9.DOI:10.16511/j.cnki.qhdxxb.2024.27.036
基于改进YOLOv8的火焰与烟雾检测算法
Fire and smoke detection algorithm based on improved YOLOv8
摘要
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,9.基金项目
国家自然科学基金民航联合研究基金(U2033206) （U2033206）
四川省重点实验室项目(MZ2022JB01) （MZ2022JB01）
航空科学基金(ASFC-20200046117001) （ASFC-20200046117001）
