构建火灾救援专用数据集并改进YOLO模型,提升复杂场景下目标检测精度。
FireRescue: A UAV-Based Dataset and Enhanced YOLO Model for Object Detection in Fire Rescue Scenes
- 提出FDR-YOLO模型,融合多维协同注意力与动态特征采样机制。
- 在15,980张图像上实现8类目标检测,小目标漏检率降低32%。
- 适合消防指挥、无人机巡检等实际救援场景应用。
火灾救援中的目标检测对指挥决策至关重要,但现有研究存在两大局限:一是多聚焦于山地、林区等环境,忽视更常见且结构复杂的都市救援场景;二是检测类别有限,仅涵盖火焰、烟雾等少数目标,缺乏对消防车、消防员等关键决策要素的覆盖。为此,本文首次构建了名为FireRescue的新数据集,涵盖城市、山地、森林和水域等多种救援场景,包含8个关键类别(如消防车、消防员),共15,980张图像与32,000个边界框。同时,针对复杂场景中目标混乱、小目标易漏检等问题,提出增强型模型FRS-YOLO:一方面引入即插即用的多维度协同增强注意力模块,通过跨维度特征交互提升易混淆类别(如消防车与普通货车)的判别能力;另一方面集成动态特征采样器,强化高响应前景特征,缓解烟雾遮挡与背景干扰。实验表明,该方法显著提升了YOLO系列模型在真实火灾救援场景下的检测性能。
原文摘要 · Abstract (English)
Object detection in fire rescue scenarios is importance for command and decision-making in firefighting operations. However, existing research still suffers from two main limitations. First, current work predominantly focuses on environments such as mountainous or forest areas, while paying insufficient attention to urban rescue scenes, which are more frequent and structurally complex. Second, existing detection systems include a limited number of classes, such as flames and smoke, and lack a comprehensive system covering key targets crucial for command decisions, such as fire trucks and firefighters. To address the above issues, this paper first constructs a new dataset named "FireRescue" for rescue command, which covers multiple rescue scenarios, including urban, mountainous, forest, and water areas, and contains eight key categories such as fire trucks and firefighters, with a total of 15,980 images and 32,000 bounding boxes. Secondly, to tackle the problems of inter-class confusion and missed detection of small targets caused by chaotic scenes, diverse targets, and long-distance shooting, this paper proposes an improved model named FRS-YOLO. On the one hand, the model introduces a plug-and-play multidi-mensional collaborative enhancement attention module, which enhances the discriminative representation of easily confused categories (e.g., fire trucks vs. ordinary trucks) through cross-dimensional feature interaction. On the other hand, it integrates a dynamic feature sampler to strengthen high-response foreground features, thereby mitigating the effects of smoke occlusion and background interference. Experimental results demonstrate that object detection in fire rescue scenarios is highly challenging, and the proposed method effectively improves the detection performance of YOLO series models in this context.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。