[发明专利]一种列车轮对踏面图像定位提取方法

摘要

本发明公开了一种列车轮对踏面图像定位提取方法，包括图像采集；图像预处理：滤波、增强并数据标注；构建基于改进的语义分割UNet网络模型；模型生成：输入经所述数据标注的数据集，对所述网络模型训练、验证，得到最佳的踏面分割模型；踏面图像定位提取：利用所述踏面分割模型对滤波和增强处理后的图像预测，得到踏面的二值化图像；利用多项式拟合技术进行踏面定位，分别得到踏面的左边界和右边界，从而提取出图像的踏面区域。本发明通过线阵相机扫描获取踏面图像，采用基于语义信息的神经网络踏面图像定位提取技术，针对车速均匀与非均匀下的情况下，能够将踏面图像自动、高效、稳定地定位和提取，提高通过式车轮踏面缺陷在线检测的鲁棒性。