A comprehensive laser image dataset for real-time measurement of wheelset geometric parameters

Abstract

Abstract Railway transportation has experienced significant growth worldwide, offering numerous benefits to society. Most railway accidents are caused by wheelset faults so it’s significant to monitor wheelset conditions. Therefore, we need to collect wheelset images, repaint them, extract laser stripe centerlines, construct 3D contour, and measure their geometric parameters to judge the wheelset’s conditions. Deep learning can fulfill the tasks satisfyingly because it’s adaptable, robust, and generalize compared with traditional methods. To train the deep learning models effectively, we need rich and high-quality wheelset datasets. So far, there are no applicable public train wheelset datasets available, which greatly hinders the research on train wheelsets. Thus we construct a publicly available Wheelset Laser Image Dataset (WLI-Set). WLI-Set consists of four sub-datasets, Original, Inpainting, Segmentation, and Centerline. The dataset contains abundant annotated multiline laser stripe images that can facilitate the research on train wheelsets effectively.