• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学

• • 上一篇    下一篇

基于改进YOLOv11n的轻量化火灾烟雾检测算法

李秦君, 于振杰, 张 童, 李 甲, 李恒越, 杨萍


  

  1. (陕西科技大学电子信息与人工智能学院,陕西 西安  710021)

Lightweight fire and smoke detection algorithm based on improved YOLOv11n

Li Qinjun, Yu Zhenjie, Zhang Tong, Li Jia, Li Hengyue, Yang Ping   

  1. (School of Electronics Information and Artificial Intelligence, Shaanxi University of Science and Technology, Xi'an 710021)

摘要: 针对边缘端火灾烟雾检测中小目标识别困难、复杂场景下精度不足以及计算冗余等问题,提出一种基于改进YOLOv11n的轻量化火灾烟雾检测算法。首先,设计PMSDA注意力模块,增强模型对小尺度目标的感知能力。其次,采用ADown下采样模块替代原始卷积结构,显著降低参数量和计算开销。此外,引入轻量且高效的动态上采样器DySample,提升特征图空间分辨率,强化细粒度特征表达能力。最后,结合紧凑倒置块CIB对部分深层结构进行优化,有效缓解计算冗余。实验结果表明,改进模型的mAP@0.5、精确率和召回率分别达到91.3%、91.1%和89.5%,较基线模型提升1.7、0.5和1.1个百分点;参数量和GFLOPs分别减少25.3%和20.6%。在GPU上推理速度达102.8FPS,嵌入式设备上达38.6FPS,且mAP@0.5达84.8%,满足边缘设备对火灾烟雾实时检测的需求。


关键词: YOLOv11n, 火灾烟雾检测, 小尺度目标检测, 轻量化网络, 计算冗余

Abstract: To address the challenges in edge-based fire and smoke detection, such as difficulties in identifying small-scale targets, low detection accuracy in complex scenes, and computational redundancy, a lightweight detection algorithm based on an improved YOLOv11n is proposed. First, a PMSDA attention module is designed to enhance the model’s perception of small-scale targets. Second, an ADown downsampling module is adopted to replace the original convolutional structure, significantly reducing the number of parameters and computational cost. Additionally, a lightweight and efficient dynamic upsampler named DySample is introduced to increase the spatial resolution of feature maps and enhance the representation of fine-grained features. Finally, a Compact Inverted Block is integrated to optimize parts of the deep layers, effectively mitigating redundant computation. Experimental results show that the improved model achieves a mAP@0.5 of 91.3%, a precision of 91.1%, and a recall of 89.5%, representing increases of 1.7, 0.5, and 1.1 percentage points, respectively, compared with the baseline model. The parameter count and GFLOPs are reduced by 25.3% and 20.6%. Deployment tests further demonstrate that the model achieves an inference speed of 102.8 FPS on GPU and 38.6 FPS on embedded devices, while maintaining a mAP@0.5 of 84.8%, satisfying the real-time detection requirements of fire and smoke monitoring on edge platforms.

Key words: YOLOv11n, fire and smoke detection, small-scale object detection, Lightweight Network, computational redundancy