arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2503.23965cs.CVcs.RO

基于视频的自动驾驶 Rockchip RV1126 交通信号灯识别

Video-based Traffic Light Recognition by Rockchip RV1126 for Autonomous Driving

NavInfo Co. Ltd.(高德信息有限公司);Autohome Inc.(汽车之家);Baidu Inc.(百度);Xidian University(西安电子科技大学)

Miao Fan, Xuxu Kong, Shengtong Xu, Haoyi Xiong, Xiangzeng Liu

AI总结:

提出基于视频的端到端神经网络 ViTLR,利用卷积自注意力架构在 Rockchip RV1126 嵌入式平台上实现鲁棒且实时的交通信号灯检测与状态分类,克服了单帧方法在复杂场景下的局限。

摘要:

实时交通信号灯识别对于自动驾驶在城市环境中的安全与导航至关重要。现有方法依赖车载摄像头的单帧分析,在遮挡和不良光照条件等复杂场景下表现不佳。我们提出了 ViTLR,一种基于视频的端到端神经网络,通过处理多个连续帧来实现鲁棒的交通信号灯检测和状态分类。该架构采用带有卷积自注意力模块的 transformer-like 设计,专门针对在 Rockchip RV1126 嵌入式平台上的部署进行了优化。在两个真实世界数据集上的广泛评估表明,ViTLR 达到了最先进的性能,同时在 RV1126 的 NPU 上保持了实时处理能力(>25 FPS)。与现有的单帧方法相比,该系统在时间稳定性、变化的目标距离以及具有挑战性的环境条件方面表现出卓越的鲁棒性。我们已成功将 ViTLR 集成到使用 HD maps 的自车道交通信号灯识别系统中,用于自动驾驶应用。包括源代码和数据集在内的完整实现已公开发布,以促进该领域的进一步研究。

英文摘要:

Real-time traffic light recognition is fundamental for autonomous driving safety and navigation in urban environments. While existing approaches rely on single-frame analysis from onboard cameras, they struggle with complex scenarios involving occlusions and adverse lighting conditions. We present \textit{ViTLR}, a novel video-based end-to-end neural network that processes multiple consecutive frames to achieve robust traffic light detection and state classification. The architecture leverages a transformer-like design with convolutional self-attention modules, which is optimized specifically for deployment on the Rockchip RV1126 embedded platform. Extensive evaluations on two real-world datasets demonstrate that \textit{ViTLR} achieves state-of-the-art performance while maintaining real-time processing capabilities (>25 FPS) on RV1126's NPU. The system shows superior robustness across temporal stability, varying target distances, and challenging environmental conditions compared to existing single-frame approaches. We have successfully integrated \textit{ViTLR} into an ego-lane traffic light recognition system using HD maps for autonomous driving applications. The complete implementation, including source code and datasets, is made publicly available to facilitate further research in this domain.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑