(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210693894.1 (22)申请日 2022.06.19 (71)申请人 成都主导科技有限责任公司 地址 610000 四川省成 都市青羊区文家路 389号 (72)发明人 张渝　赵波　彭建平　黄炜　章祥　 王小伟　马莉　彭华　韩明阳　 (74)专利代理 机构 成都市集智汇华知识产权代 理事务所(普通 合伙) 51237 专利代理师 李华　温黎娟 (51)Int.Cl. G06T 7/00(2017.01) G06V 10/82(2022.01) G06V 10/26(2022.01) G06V 10/80(2022.01)G06V 10/25(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 一种列车的车轮踏面缺陷检测方法及其系 统 (57)摘要 本发明公开了一种列车的车轮踏面缺陷检 测方法及其系统， 包括： 采集车轮踏面的二维图 像数据； 构建用于提取缺陷区域的神经网络模 型； 将所述二维图像数据输入所述神经网络模 型； 所述神经网络模型将所述二维图像数据分割 为多类图像区域； 基于多类所述图像区域提取多 个缺陷区域信息并生成检测报告。 本发明通过语 义分割网络针对频繁出现缺陷的区域进行精准 的划分， 降低图像背景干扰后， 能够通过目标检 测模型准确的提取缺陷区域信息及缺陷类型， 提 高缺陷检测准确率且降低了检测结果的漏报率 和误报率， 解决了传统的车轮踏面缺陷检测方法 存在的检测结果可靠性低的问题。 权利要求书1页 说明书3页 附图2页 CN 114998286 A 2022.09.02 CN 114998286 A 1.一种列车的车轮踏面 缺陷检测方法， 其特 征在于， 包括： 采集车轮踏面的二维图像数据； 构建用于提取缺陷区域的神经网络模型； 将所述二维图像数据输入所述神经网络模型； 所述神经网络模型将所述 二维图像数据分割为多类图像区域； 基于多类所述图像区域 提取多个缺陷区域信息并生成检测报告。 2.根据权利要求1所述的车轮踏面缺陷检测方法， 其特征在于， 构建用于提取缺陷区域 的神经网络模型， 包括： 构建初始化网络模型， 其中， 网络模型包括语义分割模型和目标检测模型； 获取包含人工标记的缺陷特征区域的车轮踏面样本 图像构成的训练数据集和测试数 据集； 基于所述训练数据集和所述测试 数据集训练并测试 所述网络模型。 3.根据权利要求2所述的车轮踏面缺陷检测方法， 其特征在于， 所述神经网络模型将所 述二维图像数据分割为多类图像区域， 包括： 所述语义分割模型基于所述 二维图像数据进行 下采样并生成多尺度的特 征图； 所述语义分割模型通过在高分辨率特征图主网络逐渐并行加入低分辨率特征图子网 络， 能够融合多个所述特 征图并生成融合后的多类所述图像区域。 4.根据权利要求3所述的车轮踏面缺陷检测方法， 其特征在于， 基于多类所述图像区域 提取多个缺陷区域信息并生成检测报告， 包括： 所述目标检测模型基于所述多类所述图像区域输出多个缺陷检测框数据构成的所述 缺陷区域信息， 其中， 所述缺陷检测框数据至少包括缺陷类型、 置信度、 检测框左上角坐标、 检测框长度和检测框 宽度。 5.根据权利要求4所述的车轮踏面缺陷检测方法， 其特征在于， 所述语义分割模型为 SegNet分割模型。 6.一种列车的车轮踏面 缺陷检测系统， 其特 征在于， 包括： 数据采集单 元， 用于采集车轮踏面的二维图像数据； 数据处理单元， 用于构建用于提取缺陷区域的神经网络模型， 所述神经网络模型将所 述二维图像数据分割为多类图像区域， 并基于多类所述图像区域提取多个缺陷区域信息并 生成检测报告。 7.根据权利要求6所述的车轮踏面缺陷检测系统， 其特征在于， 所述数据采集单元包 括： 光源组件， 用于提供摄 像单元所需的光照条件； 所述摄像单元， 用于采集所述 二维图像数据。 8.根据权利要求6所述的车轮踏面缺陷检测系统， 其特征在于， 所述车轮踏面缺陷检测 系统还包括： 数据存储单元， 用于存 储二维图像数据和样本图像。权　利　要　求　书 1/1 页 2 CN 114998286 A 2一种列车的车轮踏面缺陷检测方 法及其系统 技术领域 [0001]本发明涉及轨道车辆检测技术领域， 具体涉及一种列车的车轮踏面缺陷检测方法 及其系统。 背景技术 [0002]列车车轮是关系到列车运行安全的关键部分， 它把车辆 的载荷传给钢轨， 并在钢 轨上转动来完成列车 的运行， 是列车运行 的最终受力部件。 轮对踏面是列车车轮与钢轨顶 面接触的部 分， 轮对踏面的完好性是行车安全性的重要因素， 在运行过程中， 车轮踏面会出 现磨耗超限、 踏面擦伤、 剥离、 磕伤等轮辋表面缺陷等质量问题， 这些问题可能会直接导致 脱轨事故的发生， 影响动车组运行安全。 鉴于此， 需要对列车车轮踏面进行日常动态检测， 保障列车运行安全。 [0003]传统的国内外针对车轮踏面缺陷检测方法， 仍采用传统图像方案， 对滤波后的车 轮图像,首先利用缺陷区域与非缺陷区域之间的灰度差对缺陷区域进行粗提取,然后利用 缺陷形状规则、 边缘特征明显等特点,采用分割方案将缺陷最小外接矩形提取出来。 最后建 立正、 负缺陷样本集， 采用SVM 分类器对提取出来的缺陷图像进 行分类。 但是， 传统的缺陷检 测方法依赖图像拍摄质量， 当背景信息极其丰富的情况下， 人为设定的方案容易受到背景 信息的干扰。 造成误报警多、 检测失真情况。 [0004]综上所述， 现有的车轮踏面 缺陷检测方法存在检测结果可靠性低的问题。 发明内容 [0005]有鉴于此， 本发明提供一种列车的车轮踏面缺陷检测方法及其系统， 通过改进车 轮踏面缺陷检测方法及检测数据的处理方法， 解决了传统的车轮踏面缺陷检测方法存在的 检测结果可靠性低的问题。 [0006]为解决以上问题， 本发明的技术方案为采用一种列车的车轮踏面缺陷检测方法， 包括： 采集车轮踏面的二 维图像数据； 构建用于提取缺陷区域的神经网络模型； 将所述二 维 图像数据输入所述神经网络模型； 所述神经网络模型将所述二维图像数据分割为多类图像 区域； 基于多类所述图像区域 提取多个缺陷区域信息并生成检测报告。 [0007]可选地， 构建用于提取缺陷区域的神经网络模型， 包括： 构建初始化网络模型， 其 中， 网络模型包括语义分割模型和目标检测模型； 获取包含人工标记的缺陷特征区域的车 轮踏面样本图像构成的训练数据集和测试数据集； 基于所述训练数据集和所述测试数据集 训练并测试 所述网络模型。 [0008]可选地， 所述神经网络模型将所述二维图像数据分割为多类图像区域， 包括： 所述 语义分割模型基于所述二 维图像数据进行下采样并生成多尺度的特征图； 所述语义分割模 型通过在高分辨率特征图主网络逐渐并行加入低分辨率特征图子网络， 能够融合多个所述 特征图并生成融合后的多类所述图像区域。 [0009]可选地， 基于多类所述图像区域提取多个缺陷区域信息并生成检测报告， 包括： 所说　明　书 1/3 页 3 CN 114998286 A 3