基于改进YOLOv8的室内火情检测
Indoor Fire Detection Based on Improved YOLOv8
DOI: 10.12677/mos.2024.134439, PDF,    科研立项经费支持
作者: 杨 明:安顺开放大学,贵州 安顺;钱松荣:贵州大学省部共建公共大数据实验室,贵州 贵阳
关键词: 室内火情监测YOLOv8Ghost网络CBAM目标检测Indoor Fire Monitoring YOLOv8 Ghost Network CBAM Object Detection
摘要: 随着城市化进程的加快,室内火灾预警变得尤为重要。基于视觉的深度学习技术已成为火灾检测的研究热点,但在性能上仍有较大的改进空间。因此,本文设计了一种改良版的YOLOv8s算法,以提高室内火灾检测的准确性和实时性。改进的YOLOv8s算法通过整合Ghost模块和卷积块注意力机制(CBAM),大幅降低了计算复杂度并增强了特征融合的作用。实验结果表明,该改进模型在模型参数量方面减少了44%,同时帧率提升了19.6%,检测精度也增加了2个百分点;在与其他主流算法进行对比时,在模型的精度、召回率和参数等均体现出均衡的优势。本文详细评估了这些改进对模型检测性能的影响,结果表明,改进后的YOLOv8s算法在检测速度和准确性上均表现出显著优势。本研究不仅为室内火灾监测提供了更高效的解决方案,也展示了深度学习在火灾检测中的广阔应用前景。
Abstract: With the acceleration of urbanization, indoor fire detection has become increasingly important. Visual-based deep learning technologies have emerged as a hot research topic in fire detection, but there is still significant room for improvement in performance. Therefore, we propose an improved version of the YOLOv8s algorithm to enhance the accuracy and real-time performance of indoor fire detection. The improved YOLOv8s algorithm integrates the Ghost module and Convolutional Block Attention Module (CBAM), significantly reducing computational complexity and enhancing feature fusion. Experimental results show that the improved model reduces the number of model parameters by 44%, while increasing the frame rate by 19.6% and improving detection accuracy by 2 percentage points. When compared with other mainstream algorithms, the improved model demonstrates balanced advantages in accuracy, recall rate, and parameters. We comprehensively evaluate the impact of these improvements on model detection performance, and the results show that the improved YOLOv8s algorithm exhibits significant advantages in both detection speed and accuracy. This study not only provides a more efficient solution for indoor fire detection but also demonstrates the broad application prospects of deep learning in fire detection.
文章引用:杨明, 钱松荣. 基于改进YOLOv8的室内火情检测[J]. 建模与仿真, 2024, 13(4): 4863-4871. https://doi.org/10.12677/mos.2024.134439

参考文献

[1] Hussain, M. (2023) YOLO-v1 to YOLO-v8, the Rise of YOLO and Its Complementary Nature toward Digital Manufacturing and Industrial Defect Detection. Machines, 11, 677. [Google Scholar] [CrossRef
[2] Han, K., Wang, Y., Tian, Q., Guo, J., Xu, C. and Xu, C. (2020) GhostNet: More Features from Cheap Operations. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, 13-19 June 2020, 1577-1586. [Google Scholar] [CrossRef
[3] Woo, S., Park, J., Lee, J. and Kweon, I.S. (2018) CBAM: Convolutional Block Attention Module. Computer Vision-ECCV 2018, Munich, 8-14 September 2018, 3-19. [Google Scholar] [CrossRef
[4] Yang, Q., Yu, F., Zhang, G., Guo, D., Wang, P. and Yao, G. (2023) Research on Image-Based Wildfire Intelligent Detection Method. International Conference on Artificial Intelligence, Virtual Reality, and Visualization (AIVRV 2022), Chongqing, 1 March 2023, 125880Z. [Google Scholar] [CrossRef
[5] Zhang, Z., Jin, Q., Wang, L. and Liu, Z. (2022) Video-Based Fire Smoke Detection Using Temporal-Spatial Saliency Features. Procedia Computer Science, 198, 493-498. [Google Scholar] [CrossRef
[6] Majid, S., Alenezi, F., Masood, S., Ahmad, M., Gündüz, E.S. and Polat, K. (2022) Attention Based CNN Model for Fire Detection and Localization in Real-World Images. Expert Systems with Applications, 189, Article 116114. [Google Scholar] [CrossRef
[7] Muhammad, K., Ahmad, J., Lv, Z., Bellavista, P., Yang, P. and Baik, S.W. (2019) Efficient Deep CNN-Based Fire Detection and Localization in Video Surveillance Applications. IEEE Transactions on Systems, Man, and Cybernetics: Systems, 49, 1419-1434. [Google Scholar] [CrossRef
[8] Mseddi, W.S., Ghali, R., Jmal, M. and Attia, R. (2021) Fire Detection and Segmentation Using YOLOv5 and U-Net. 2021 29th European Signal Processing Conference (EUSIPCO), Dublin, 23-27 August 2021, 741-745. [Google Scholar] [CrossRef
[9] Terven, J., Córdova-Esparza, D.-M. and Romero-González, J.-A. (2023) A Comprehensive Review of YOLO Architectures in Computer Vision: From YOLOv1 to YOLOv8 and YOLO-NAS. Machine Learning and Knowledge Extraction, 5, 1680-1716. [Google Scholar] [CrossRef
[10] Khalid, S., Oqaibi, H.M., Aqib, M. and Hafeez, Y. (2023) Small Pests Detection in Field Crops Using Deep Learning Object Detection. Sustainability, 15, Article 6815. [Google Scholar] [CrossRef
[11] Wang, J., Dai, H., Chen, T., Liu, H., Zhang, X., Zhong, Q., et al. (2023) Toward Surface Defect Detection in Electronics Manufacturing by an Accurate and Lightweight YOLO-Style Object Detector. Scientific Reports, 13, Article No. 7062. [Google Scholar] [CrossRef] [PubMed]
[12] Wu, Y., Gao, D., Fang, Y., Xu, X., Gao, H. and Ju, Z. (2023) SDE-YOLO: A Novel Method for Blood Cell Detection. Biomimetics, 8, Article 404. [Google Scholar] [CrossRef] [PubMed]
[13] Wang, S., Li, Y. and Qiao, S. (2024) ALF-YOLO: Enhanced YOLOv8 Based on Multiscale Attention Feature Fusion for Ship Detection. Ocean Engineering, 308, Article 118233. [Google Scholar] [CrossRef
[14] Zhang, S. and Ge, J. (2023) Vehicle Detection Algorithm for Highway Driving Scenarios. Fifth International Conference on Artificial Intelligence and Computer Science (AICS 2023), Wuhan, 16 October 2023, 128032O. [Google Scholar] [CrossRef
[15] Pan, P., Guo, W., Zheng, X., Hu, L., Zhou, G. and Zhang, J. (2023) Xoo-YOLO: A Detection Method for Wild Rice Bacterial Blight in the Field from the Perspective of Unmanned Aerial Vehicles. Frontiers in Plant Science, 14, Article 1256545. [Google Scholar] [CrossRef] [PubMed]
[16] Dalal, S., Lilhore, U.K., Radulescu, M., Simaiya, S., Jaglan, V. and Sharma, A. (2024) A Hybrid LBP-CNN with YOLO-v5-Based Fire and Smoke Detection Model in Various Environmental Conditions for Environmental Sustainability in Smart City. Environmental Science and Pollution Research. [Google Scholar] [CrossRef] [PubMed]
[17] Niu, Z., Zhong, G. and Yu, H. (2021) A Review on the Attention Mechanism of Deep Learning. Neurocomputing, 452, 48-62. [Google Scholar] [CrossRef
[18] Han, J. and Moraga, C. (1995) The Influence of the Sigmoid Function Parameters on the Speed of Backpropagation Learning. In: Mira, J. and Sandoval, F., Eds., From Natural to Artificial Neural Computation, Springer Berlin Heidelberg, 195-201. [Google Scholar] [CrossRef
[19] Zhu, X., Cheng, D., Zhang, Z., Lin, S. and Dai, J. (2019) An Empirical Study of Spatial Attention Mechanisms in Deep Networks. 2019 IEEE/CVF International Conference on Computer Vision (ICCV), Seoul, 27 October-2 November 2019, 6687-6696. [Google Scholar] [CrossRef
[20] 孙孟凡. 面向图像理解的轻量级网络关键技术研究与原型实现[D]: [硕士学位论文]. 成都: 电子科技大学, 2022.
[21] 崔志亮, 曹苏群. 基于改进YOLOv5的复杂环境下火灾检测方法[J]. 电脑与信息技术, 2024, 32(1): 23-27, 46.
[22] 雷景生, 李顶, 俞云祥, 等. 基于改进YOLOv7的火焰烟雾识别[J/OL]. 计算机仿真, 1-7.
http://kns.cnki.net/kcms/detail/11.3724.TP.20240327.1720.008.html, 2024-05-20.