arXiv:2510.16854cs.CVcs.AI2025-10被引 1

轻量级视觉模型实现武器实时精准分割,兼顾速度与精度。

ArmFormer: Lightweight Transformer Architecture for Real-Time Multi-Class Weapon Segmentation and Classification

  • 结合CBAM与MixVisionTransformer构建轻量编码器,提升特征提取效率。
  • 在五类武器上达80.64% mIoU和89.13% mFscore,推理速度82.26 FPS。
  • 仅需4.886G FLOPs,适合部署于边缘设备如安防摄像头、无人机。

武器暴力威胁加剧,亟需具备像素级精度的实时检测系统以支持精准威胁评估。传统目标检测方法仅提供粗略边界框,无法满足精细分割需求;现有语义分割模型或牺牲精度换取效率,或计算开销过大,难以在边缘设备部署。本文提出ArmFormer,一种基于轻量级Transformer的语义分割框架,通过将卷积块注意力模块(CBAM)与MixVisionTransformer架构相结合,在保持高效计算的同时实现高精度。该模型采用增强编码器与注意力融合的汉堡解码器,支持手枪、步枪、刀具、左轮、人五类武器的多类别分割。实验表明,ArmFormer在保持82.26 FPS实时推理速度下,达到80.64% mIoU和89.13% mFscore,仅需4.886G FLOPs和3.66M参数,远优于需48倍以上计算资源的重型模型,是便携式安防相机、监控无人机及分布式安全系统中嵌入式AI加速器的理想选择。

原文摘要 · Abstract (English)

The escalating threat of weapon-related violence necessitates automated detection systems capable of pixel-level precision for accurate threat assessment in real-time security applications. Traditional weapon detection approaches rely on object detection frameworks that provide only coarse bounding box localizations, lacking the fine-grained segmentation required for comprehensive threat analysis. Furthermore, existing semantic segmentation models either sacrifice accuracy for computational efficiency or require excessive computational resources incompatible with edge deployment scenarios. This paper presents ArmFormer, a lightweight transformer-based semantic segmentation framework that strategically integrates Convolutional Block Attention Module (CBAM) with MixVisionTransformer architecture to achieve superior accuracy while maintaining computational efficiency suitable for resource-constrained edge devices. Our approach combines CBAM-enhanced encoder backbone with attention-integrated hamburger decoder to enable multi-class weapon segmentation across five categories: handgun, rifle, knife, revolver, and human. Comprehensive experiments demonstrate that ArmFormer achieves state-of-the-art performance with 80.64% mIoU and 89.13% mFscore while maintaining real-time inference at 82.26 FPS. With only 4.886G FLOPs and 3.66M parameters, ArmFormer outperforms heavyweight models requiring up to 48x more computation, establishing it as the optimal solution for deployment on portable security cameras, surveillance drones, and embedded AI accelerators in distributed security infrastructure.

图像分割轻量模型边缘计算安防应用

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