arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.14366cs.CV
收藏

Rail-5 k:用于钢轨表面缺陷检测的真实数据集

Rail-5k: a Real-World Dataset for Rail Surface Defects Detection

Tongji University(同济大学)

Zihao Zhang, Shaozuo Yu, Siwei Yang, Yu Zhou, Bingchen Zhao

摘要:

本文介绍了Rail-5 k数据集,用于在现实世界的应用场景中对视觉算法的性能进行基准测试,即轨道表面缺陷检测任务。我们从中国各地的铁路收集了超过5000张高质量的图像,并在铁路专家的帮助下对1100张图像进行了注释,以识别最常见的13种钢轨缺陷。该数据集可用于两种具有独特挑战的设置,第一种是使用1 k+标记图像进行训练的全监督设置,细粒度性质和缺陷类的长尾分布使得视觉算法难以处理。第二种是由4k幅未标记图像促进的半监督学习设置,这些4k幅图像是未策划的,包含可能的图像损坏和与标记图像的域偏移,这不能通过先前的半监督学习方法容易地解决。我们相信我们的数据集可以成为评估视觉算法的鲁棒性和可靠性的有价值的基准。

英文摘要:

This paper presents the Rail-5k dataset for benchmarking the performance of visual algorithms in a real-world application scenario, namely the rail surface defects detection task. We collected over 5k high-quality images from railways across China, and annotated 1100 images with the help from railway experts to identify the most common 13 types of rail defects. The dataset can be used for two settings both with unique challenges, the first is the fully-supervised setting using the 1k+ labeled images for training, fine-grained nature and long-tailed distribution of defect classes makes it hard for visual algorithms to tackle. The second is the semi-supervised learning setting facilitated by the 4k unlabeled images, these 4k images are uncurated containing possible image corruptions and domain shift with the labeled images, which can not be easily tackle by previous semi-supervised learning methods. We believe our dataset could be a valuable benchmark for evaluating robustness and reliability of visual algorithms.

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

↑