基于改进YOLOv7的多场景火灾识别算法

Multi-Scenario Fire Detection Algorithm Based on Improved YOLOv7

  • 摘要: 为解决传统火警报警系统在火灾情况下由于环境和空间限制导致检测效率下降的问题,本研究提出了一种针对多环境火灾检测改进的YOLOv7算法。在原算法中集成动态头(dynamic head)提升检测精度,并引入全局注意力机制(global attention mechanism, GAM)与Slim-Neck模型结构以增强对火灾特征的提取能力和降低算法的计算量。此外,采用Wise-IoUv3损失函数代替传统模型的IoU损失函数,针对不同画质图像调整训练权重,优化模型效率。实验结果表明,在多环境下的火灾数据集上,改进后的YOLOv7算法mAP@0.5达到71.1%,较原始算法提升了5.3个百分点,展现出极高的检测精度,有效满足复杂环境下精确火灾检测的需求。

     

    Abstract: To address the issue of reduced detection efficiency in traditional fire alarm systems owing to environmental and spatial limitations during fire incidents, this study investigated an improved YOLOv7 algorithm specifically designed for multi-environment fire detection. A dynamic head was integrated into the original algorithm to enhance the detection accuracy, and a global attention mechanism and slim neck model structure were introduced to improve the extraction of fire-related features and reduce the computational complexity. Additionally, the Wise-IoUv3 loss function was employed instead of the traditional IoU loss function in the model, and the training weights were adjusted based on the image quality to optimize the model efficiency. In an experiment using a multi-environment fire dataset, the improved YOLOv7 algorithm achieved an mAP@0.5 of 71.1%, which was a 5.3 percent improvement over the result with the original algorithm. This demonstrated a high level of detection accuracy and effectively met the requirements for precise fire detection in complex environments.

     

/

返回文章
返回