欢迎光临散文网 会员登陆 & 注册

CDSet: 开源斑马线检测数据集和相应开源代码

2023-08-29 00:09 作者:白小心0225  | 我要投稿

CDSet数据集(Crosswalk DataSet)是在车载摄像头上采集到的斑马线图像,用于研究先进的斑马线目标检测和汽车过线行为分析的神经网络算法。


CDNet实时斑马线检测神经网络


CDSet数据集包含3434张图像,包括白天、雨天、遮挡、变形、截断、夜晚、破损、炫光等真实场景,目标包含斑马线和导流线两个类别,数据集被切分为3080张训练集和354张测试集。此外,本数据集还提供了额外的标注为有无斑马线的1770张图像用于测试。

对应论文`CDNet: a real-time and robust crosswalk detection network on Jetson nano based on YOLOv5`,点此阅读论文:https://rdcu.be/cHuc8, DOI: 10.1007/s00521-022-07007-9。CDNet基于当时的SOTA模型YOLOv5并进行了ROI、SE、NST、SSVM、Synthetic Fog等多项改进,在Nvidia Jetson Nano上实现了33.1FPS和94.83% F1分数的实时和鲁棒效果。开源代码见https://github.com/zhangzhengde0225/CDNet.

斑马线检测效果

数据集和代码均可免费用于学术研究。

CDSet: 开源斑马线检测数据集和相应开源代码的评论 (共 条)

分享到微博请遵守国家法律