<p>Real-time and accurate detection of dangerous weapons is crucial for preventing violent crimes and maintaining social stability. Although significant progress has been made in deep learning-based weapon detection, these methods still face considerable challenges in multi-source complex scenarios. Traditional models often suffer from inefficient feature utilization, poor decoupling of weapon subclasses, and inadequate feature fusion under domain shifts. Such domain shift problems lead to significant performance decline in complex environments. To address these limitations, this paper proposes the Ubiquitous Arms Incidents (UAI) dataset and RW-DETR, a weapon detection framework based on a multi-domain fusion transformer. By leveraging multi-source scenario styles, UAI enhances data diversity and effectively mitigates the data domain shift. RW-DETR employs an Efficient Dynamic Gated Backbone (EDG-Backbone), combined with conditional positional encoding and a dynamic focus mechanism, to enhance feature capture in key domains. To enhance fine-grained feature discrimination in weapon subclasses, we propose the Aurora PrismFormer Layer (APFL), which leverages lightweight linear mapping and bipolar filtering to precisely differentiate subtle inter-class variations. Furthermore, we propose a progressive contextual reshaping fusion mechanism that utilizes flexible multi-scale feature interactions to significantly enhance adaptability to targets of varying scales. Experimental results demonstrate that RW-DETR achieves 84.6% AP<InlineEquation ID="IEq1"> <EquationSource Format="TEX">\(_{50}\)</EquationSource> <EquationSource Format="MATHML"><math> <mmultiscripts> <mrow /> <mn>50</mn> <mrow /> </mmultiscripts> </math></EquationSource> </InlineEquation> and 69.0% AP<InlineEquation ID="IEq2"> <EquationSource Format="TEX">\(_{50-95}\)</EquationSource> <EquationSource Format="MATHML"><math> <mmultiscripts> <mrow /> <mrow> <mn>50</mn> <mo>-</mo> <mn>95</mn> </mrow> <mrow /> </mmultiscripts> </math></EquationSource> </InlineEquation> on the UAI dataset with 7.3ms inference, demonstrating robust performance across three public datasets. Relative to the baseline, it reduces parameters by 36.5% and GFLOPs by 38.6%. Notably, RW-DETR outperforms mainstream models in complex occlusion and low-light scenarios, delivering superior accuracy while ensuring real-time performance.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

RW-DETR: a multi-domain fusion transformer for real-time weapon detection in public security

  • Yuan Jianan,
  • Ding Meng,
  • Xu Xiaoyu,
  • He Yilin,
  • Ma Wanli,
  • Gu Hongxing,
  • Yu Jiangfeng,
  • Li Qing

摘要

Real-time and accurate detection of dangerous weapons is crucial for preventing violent crimes and maintaining social stability. Although significant progress has been made in deep learning-based weapon detection, these methods still face considerable challenges in multi-source complex scenarios. Traditional models often suffer from inefficient feature utilization, poor decoupling of weapon subclasses, and inadequate feature fusion under domain shifts. Such domain shift problems lead to significant performance decline in complex environments. To address these limitations, this paper proposes the Ubiquitous Arms Incidents (UAI) dataset and RW-DETR, a weapon detection framework based on a multi-domain fusion transformer. By leveraging multi-source scenario styles, UAI enhances data diversity and effectively mitigates the data domain shift. RW-DETR employs an Efficient Dynamic Gated Backbone (EDG-Backbone), combined with conditional positional encoding and a dynamic focus mechanism, to enhance feature capture in key domains. To enhance fine-grained feature discrimination in weapon subclasses, we propose the Aurora PrismFormer Layer (APFL), which leverages lightweight linear mapping and bipolar filtering to precisely differentiate subtle inter-class variations. Furthermore, we propose a progressive contextual reshaping fusion mechanism that utilizes flexible multi-scale feature interactions to significantly enhance adaptability to targets of varying scales. Experimental results demonstrate that RW-DETR achieves 84.6% AP \(_{50}\) 50 and 69.0% AP \(_{50-95}\) 50 - 95 on the UAI dataset with 7.3ms inference, demonstrating robust performance across three public datasets. Relative to the baseline, it reduces parameters by 36.5% and GFLOPs by 38.6%. Notably, RW-DETR outperforms mainstream models in complex occlusion and low-light scenarios, delivering superior accuracy while ensuring real-time performance.