arXiv:2409.14852cs.CVcs.AI2024-09被引 1

FUSED-Net用少量数据实现交通标志高精度检测,适合数据稀缺场景。

FUSED-Net: Detecting Traffic Signs with Limited Data

  • 全参数不冻结+伪支持集增强,从少样本中学习
  • 1次、3次、5次、10次样本下mAP提升1.3至2.4倍
  • 适用于跨域小样本检测,尤其适合数据难收集的国家

自动驾驶交通标志识别对现代交通系统至关重要,但各国标志外观差异大,大规模数据集难以构建。为此提出FUSED-Net,基于Faster RCNN改进,引入未冻结参数、伪支持集、嵌入归一化与领域自适应,显著降低数据依赖。训练时保持所有参数可调,利用数据增强生成伪支持集弥补目标域数据不足。嵌入归一化减少类内差异,提升特征一致性。通过在非目标域预训练实现领域自适应,增强泛化能力。在BDTSD数据集上,1/3/5/10次样本下,mAP分别较当前最优少样本检测模型提升2.4倍、2.2倍、1.5倍和1.3倍。同时在跨域少样本检测任务中也优于现有方法。

原文摘要 · Abstract (English)

Automatic Traffic Sign Recognition is paramount in modern transportation systems, motivating several research endeavors to focus on performance improvement by utilizing large-scale datasets. As the appearance of traffic signs varies across countries, curating large-scale datasets is often impractical; and requires efficient models that can produce satisfactory performance using limited data. In this connection, we present 'FUSED-Net', built-upon Faster RCNN for traffic sign detection, enhanced by Unfrozen Parameters, Pseudo-Support Sets, Embedding Normalization, and Domain Adaptation while reducing data requirement. Unlike traditional approaches, we keep all parameters unfrozen during training, enabling FUSED-Net to learn from limited samples. The generation of a Pseudo-Support Set through data augmentation further enhances performance by compensating for the scarcity of target domain data. Additionally, Embedding Normalization is incorporated to reduce intra-class variance, standardizing feature representation. Domain Adaptation, achieved by pre-training on a diverse traffic sign dataset distinct from the target domain, improves model generalization. Evaluating FUSED-Net on the BDTSD dataset, we achieved 2.4x, 2.2x, 1.5x, and 1.3x improvements of mAP in 1-shot, 3-shot, 5-shot, and 10-shot scenarios, respectively compared to the state-of-the-art Few-Shot Object Detection (FSOD) models. Additionally, we outperform state-of-the-art works on the cross-domain FSOD benchmark under several scenarios.

少样本检测交通标志小样本学习目标检测

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