融合自适应注意力的多尺度火灾检测算法

Multi-Scale Fire Detection Algorithm with Adaptive Attention

  • 摘要: 针对实际场景下火灾检测准确度不高的现象,根据火灾自身特征提出了一种基于Anchor-free结构的火灾检测算法. 将特征提取网络残差模块设计为多分支结构,并在其基础上嵌入根据火焰特征设计的自适应注意力模块,提取出更具表达力的火焰特征. 添加亚像素融合,利用高层特征丰富的通道信息增强多尺度特征的表达能力. 设计特征增强模块强化最高层特征表示,更好地利用全局空间信息. 引入自适应标签分配,强化网络对火焰特征的学习效果. 利用改进后的GIoU Loss损失函数对边界框精细回归. 该算法在自建数据集上的检测精度达到了94.9%,在公开数据集上也有较好的检测效果,且抗干扰能力强,适用于各种环境下的火灾检测,能够满足实际场景下火灾检测任务的需要.

     

    Abstract: In order to improve the accuracy of fire detection in actual scenes, a fire detection algorithm was proposed based on Anchor-free structure according to the characteristics of fire. Firstly, the ResNet block of feature extraction network was designed as a multi-branch structure, and the attention block designed according to the flame features was embedded on the basis to extract the more expressive features. And then, a subpixel fusion was added to enhance the expression ability of multi-scale features by utilizing the abundant feature information in high-level channel. Finally, a feature enhancement module was designed to enhance the top-level feature representation and make better use of global spatial information. Adaptive label assignment was introduced to enhance the learning ability of the extraction network to flame features. Retreating carefully to the boundary condition with improved GIoU Loss function, the detection accuracy of this algorithm can reach up 94.9% on the self-built data set, also show a good detection effect on the public data set. The experimental results show that the algorithm model can provide a high detection accuracy and strong anti-interference ability. In addition, this algorithm can also provide a better detection effect on multi-scale flame under complex background, suitable for fire detection in various environments, can meet the needs of actual fire detection tasks.

     

/

返回文章
返回