计算机与现代化 ›› 2025, Vol. 0 ›› Issue (06): 79-85.doi: 10.3969/j.issn.1006-2475.2025.06.013
摘要: 摘要:针对复杂场景下的火灾图像中出现的火焰烟雾部分被遮挡、火焰烟雾与背景难分辨导致检测精度低等问题,提出一种检测模型CFD-YOLO(Complex Fire Detection YOLO)。该模型以YOLOv8模型为基准框架,首先,在主干网络部分将可变形卷积DCNv4与C2f模块相结合,利用DCNv4的无界动态权重机制,显著提升了对火灾图像中复杂形变和非刚性特征的捕捉能力;其次,在Neck部分引入了基于交叉注意力的深层语义融合模块PSFM,通过对火灾图像不同特征层次进行深度语义融合,实现自适应的特征增强;最后,在Head部分,通过遮挡感知注意力SEAM对检测头进行改进,得到能够识别遮挡的检测头,有效改善复杂环境中火焰和烟雾被遮挡的问题;损失函数使用可动态调节正负样本系数的SlideLoss,降低误检率。在自建数据集和公开数据集下的实验结果表明,mAP指标分别达到了80.33%、88.28%,相较于原YOLOv8网络分别提升了3.85、3.91个百分点,与当前主流模型相比也具有良好的检测性能。
中图分类号: