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