用点标注+少量框标注,实现X光违禁品高精度检测
BCR-Net: Boundary-Category Refinement Network for Weakly Semi-Supervised X-Ray Prohibited Item Detection with Points
- 基于分组R-CNN,设计边界与类别双重精修模块
- 在有限标注下,检测精度超越当前最佳方法
- 适合安检场景中标注成本敏感的检测任务
X光图像中的违禁物品自动检测对公共安全至关重要。现有方法要么依赖昂贵的框标注以获得高性能,要么使用弱标注但准确率受限。为平衡标注成本与检测性能,本文研究仅需少量框标注和大量点标注的弱半监督X光违禁品检测(WSSPID-P),提出新型边界-类别精修网络(BCR-Net)。该模型基于Group R-CNN,引入边界精修(BR)模块和类别精修(CR)模块。BR模块采用双注意力机制,聚焦违禁物品的边界与显著特征;CR模块在RPN和ROI头部引入尺度与旋转感知对比损失,增强特征空间内类内一致性与类间可分性。实验表明,该方法在公开X光数据集上显著优于现有技术,在标注受限条件下实现性能突破。
原文摘要 · Abstract (English)
Automatic prohibited item detection in X-ray images is crucial for public safety. However, most existing detection methods either rely on expensive box annotations to achieve high performance or use weak annotations but suffer from limited accuracy. To balance annotation cost and detection performance, we study Weakly Semi-Supervised X-ray Prohibited Item Detection with Points (WSSPID-P) and propose a novel \textbf{B}oundary-\textbf{C}ategory \textbf{R}efinement \textbf{Net}work (\textbf{BCR-Net}) that requires only a few box annotations and a large number of point annotations. BCR-Net is built based on Group R-CNN and introduces a new Boundary Refinement (BR) module and a new Category Refinement (CR) module. The BR module develops a dual attention mechanism to focus on both the boundaries and salient features of prohibited items. Meanwhile, the CR module incorporates contrastive branches into the heads of RPN and ROI by introducing a scale- and rotation-aware contrastive loss, enhancing intra-class consistency and inter-class separability in the feature space. Based on the above designs, BCR-Net effectively addresses the closely related problems of imprecise localization and inaccurate classification. Experimental results on public X-ray datasets show the effectiveness of BCR-Net, achieving significant performance improvements to state-of-the-art methods under limited annotations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。