arXiv:2509.23697cs.CVcs.LG2025-09被引 1

用多模型融合提升枪支检测准确率,尤其在遮挡和光照复杂场景下表现更优。

Confidence Aware SSD Ensemble with Weighted Boxes Fusion for Weapon Detection

  • 集成不同骨干网络的SSD模型,增强特征表达多样性。
  • 加权框融合(WBF)结合置信度策略,使mAP达0.838,优于单模型2.948%。
  • 适合需高鲁棒性实时安防检测的场景,如公共区域监控系统。

公共空间的安全至关重要,推动了对精确武器检测系统的需求,但此类检测常受部分遮挡、光照变化和背景杂乱等问题影响。尽管单模型检测器性能先进,但在复杂条件下仍缺乏鲁棒性。本文提出,采用不同特征提取骨干网络的单阶段多框检测器(SSD)集成可显著提升检测鲁棒性。为此,使用VGG16、ResNet50、EfficientNet和MobileNetV3作为骨干网络训练多个SSD模型。实验基于包含三类武器(枪支、重型武器、刀具)的图像数据集进行。通过加权框融合(WBF)方法整合各模型预测结果,该方法旨在优化边界框精度。关键发现表明,融合策略与集成多样性同等重要;采用'最大置信度'评分策略的WBF方法达到0.838的平均精度均值(mAP),相比表现最佳的单模型提升2.948%,且持续优于其他融合启发式方法。研究证明,置信度感知融合是提升集成模型精度的关键机制,为实时安防监控中的武器检测提供了可靠方案。

原文摘要 · Abstract (English)

The safety and security of public spaces is of vital importance, driving the need for sophisticated surveillance systems capable of accurately detecting weapons, which are often hampered by issues like partial occlusion, varying lighting, and cluttered backgrounds. While single-model detectors are advanced, they often lack robustness in these challenging conditions. This paper presents the hypothesis that ensemble of Single Shot Multibox Detector (SSD) models with diverse feature extraction backbones can significantly enhance detection robustness. To leverage diverse feature representations, individual SSD models were trained using a selection of backbone networks: VGG16, ResNet50, EfficientNet, and MobileNetV3. The study is conducted on a dataset consisting of images of three distinct weapon classes: guns, heavy weapons and knives. The predictions from these models are combined using the Weighted Boxes Fusion (WBF) method, an ensemble technique designed to optimize bounding box accuracy. Our key finding is that the fusion strategy is as critical as the ensemble's diversity, a WBF approach using a 'max' confidence scoring strategy achieved a mean Average Precision (mAP) of 0.838. This represents a 2.948% relative improvement over the best-performing single model and consistently outperforms other fusion heuristics. This research offers a robust approach to enhancing real-time weapon detection capabilities in surveillance applications by demonstrating that confidence-aware fusion is a key mechanism for improving accuracy metrics of ensembles.

目标检测武器识别模型集成WBF融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。