Command Palette

Search for a command to run...

3 个月前

一种面向小尺寸交通标志识别的实时高精度方法

{Ronghui Zhang Zhihan Lv Wenquan Chen Kunkun Jia Junzhou Chen}

摘要

作为交通系统的基本构成要素,交通标志通过向驾驶员、行人等提供道路状况的关键信息,有效降低事故发生的风险。随着计算机视觉与人工智能技术的迅猛发展,交通标志识别系统已被广泛应用于高级驾驶辅助系统(ADAS)及自动驾驶系统中,以帮助驾驶员和自动驾驶车辆精准获取重要道路信息。然而,在实际应用中,小尺寸交通标志的识别仍面临较大挑战。本文提出一种高效的小尺寸交通标志识别方法——Small-Aware Traffic Signs Recognition,该方法受当前先进目标检测框架YOLOv4与YOLOv5的启发。本研究主要包含以下四项贡献:(1)在模型主干网络(Backbone)方面,引入高层语义特征,构建更优的检测头;(2)在模型颈部结构(Neck)中,采用感受野增强模块(Receptive Field BlockCross),以更好地捕捉特征图的上下文信息;(3)在模型检测头(Head)部分,优化检测网格结构,提升对小尺寸交通标志的定位精度;(4)在输入处理方面,提出一种名为“随机擦除-注意力”(Random Erasing-Attention)的数据增强方法,能够有效增加难样本数量,增强模型的鲁棒性。在具有挑战性的TT100K数据集上的实验证明,所提方法在性能上显著优于现有最先进方法。此外,该方法具备实时处理能力,展现出在高级驾驶辅助系统与自动驾驶系统中广阔的应用前景。

基准测试

基准方法指标
traffic-sign-recognition-on-tsinghua-tencentTSR-SA(without RFB-C)
FPS (V100, b=1): 58.1
MAP: 0.899
traffic-sign-recognition-on-tsinghua-tencentTSR-SA(with RFB-C)
FPS (V100, b=1): 48.8
MAP: 0.902

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供