关键词:
火灾检测
YOLOv5
注意力模块
多分支卷积
摘要:
预警火灾对人类生命财产安全具有重要意义。为了提高算法在实际火灾场景中的检测能力,文章提出了一种基于YOLOv5改进的火灾检测算法YOLO-IFD。在YOLOv5网络的基础上进行改进,首先在主干网络中嵌入Shuffle模块,通过学习通道特征和空间特征,保持高分辨率信息,从而抑制冗余特征,同时增强全局和局部特征的提取能力;其次,引入一种新的残差双重注意力模块(RDAM),进一步增强关键特征的表达能力;最后,在网络结构的最深层特征后添加一个多分支卷积模块(RFB),该模块能够融合不同尺度的空间语义信息和细节信息,有效分离出最显著的上下文特征,提升检测精度。实验结果表明,YOLO-IFD相比YOLOv5s原网络,平均精度(mAP)提升了6.1%,且检测速度达到了121fps,能够满足实际场景中的实时火灾检测需求。