3 个月前

CCSPNet-Joint:极端条件下交通标志检测的高效联合训练方法

CCSPNet-Joint:极端条件下交通标志检测的高效联合训练方法

摘要

交通标志检测是智能驾驶领域的重要研究方向。然而,现有方法往往忽视了雾天、雨天及运动模糊等极端环境条件。此外,图像去噪与目标检测模型的端到端训练策略难以有效利用模型间的相互信息。针对上述问题,本文提出CCSPNet——一种基于上下文变换器(Contextual Transformer)与卷积神经网络(CNN)的高效特征提取模块,能够有效融合图像的静态与动态特征,实现更快的推理速度,并具备更强的特征增强能力。进一步地,我们揭示了目标检测与图像去噪任务之间的内在关联,提出一种联合训练模型CCSPNet-Joint,以提升数据利用效率与模型泛化能力。最后,为验证所提方法的有效性,我们构建了面向极端场景下交通标志检测的CCTSDB-AUG数据集。大量实验结果表明,CCSPNet在极端条件下的交通标志检测任务中达到了当前最优性能;相较于端到端方法,CCSPNet-Joint在精度上提升5.32%,在mAP@0.5指标上提升18.09%。

代码仓库

基准测试

基准方法指标
traffic-sign-detection-on-cctsdb-augYOLO-CCSPNet
Averaged Precision: 0.917
avg-mAP (0.1-0.5): 0.861
traffic-sign-detection-on-cctsdb-augCCSPNet-Joint
Averaged Precision: 0.951
avg-mAP (0.1-0.5): 0.914
traffic-sign-detection-on-cctsdb2021YOLO-CCSPNet
mAP@0.5: 95.8

用 AI 构建 AI

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

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

Hyper Newsletters

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