一种基于深度学习技术的机车车辆闸片磨耗检测方法技术

本发明专利技术公开了一种基于深度学习技术的机车车辆闸片磨耗检测方法。现有闸片图像检测技术存在准确度低以及场景适应性差的问题。本发明专利技术通过设计AI语义分割网络，获得闸片的像素点边缘坐标及闸片类型；利用亚像素边缘提取算法，得到亚像素边缘坐标；将亚像素边缘坐标输入标定计算模块，并基于概率统计的异常值剔除算法，得到闸片厚度。本发明专利技术基于人工智能算法，算法适应性好，可以适应不同的光照、不同的闸片类型，检测准确率高，便于推广应用。便于推广应用。便于推广应用。

【技术实现步骤摘要】
一种基于深度学习技术的机车车辆闸片磨耗检测方法


[0001]本专利技术属于计算机检测
，特别涉及一种基于深度学习技术的机车车辆闸片磨耗检测方法。

技术介绍

[0002]闸片为机车车辆制动系统的核心部件，当闸片磨损到一定程度就会失效，影响机车车辆的制动性能和运行安全，因此需对机车车辆闸片厚度进行定期检测和更换。人工测量存在劳动强度大、操作不便、效率低、检测准确度低和数据误读等问题，且易发生闸片超限服役或提前更换的现象。
[0003]基于非接触式传统图像算法进行测量的优势在于不再依赖于人工，而是通过图像算法进行自动检测，可以保障测量一致性；但不同的车型会装不同类型的闸片，即便是同一种车型，也存在安装的闸片类型不统一的情况；传统的图像算法在闸片定位问题上可以使用模板匹配、特征提取、特征匹配等技术，针对闸片厚度分割问题上可以采用基于阈值、区域生长、区域分裂合并等技术，但由于光照、污水、油渍等干扰，对图像检测技术造成很大影响，同时算法的检测正确率不高。

技术实现思路

[0004]本专利技术提供一种基于深度学习技术的机车车辆闸片磨耗检测的方法，解决传统图像检测技术准确度低以及场景适应性差的问题。
[0005]为了达到上述目的，本专利技术所采用的技术方案为：
[0006]一种基于深度学习技术的机车车辆闸片磨耗检测方法，具体包括如下步骤：
[0007]步骤一：根据检测需求，通过AI目标检测网络定义闸片磨耗体区域，并对闸片进行数据标注；
[0008]步骤二：设计AI语义分割网络，利用所述闸片标注数据进行模型训练，并保存训练好的模型；
[0009]步骤三：设备采集到的二维闸片图像数据通过无线或者有线的方式，传输至AI闸片语义分割网络，所述AI闸片语义分割网络输出闸片的像素点边缘坐标及闸片类型；
[0010]步骤四：利用亚像素边缘提取算法，得到所述二维闸片图像数据像素点边缘坐标的亚像素边缘坐标；
[0011]步骤五：将所述二维闸片图像数据的亚像素边缘坐标输入标定计算模块；所述标定计算模块分离出所述二维闸片图像数据亚像素边缘坐标中左右边缘匹配的坐标点，组成直线段；所述标定计算模块通过已经标定好的参数，将所述直线段的像素长度转换成物理长度；最后基于概率统计的异常值剔除算法，剔除所述直线段中的异常直线段，得到所有保留直线段的长度平均值即为闸片厚度；
[0012]步骤六：将所述闸片厚度输入闸片厚度超限报警模块，根据报警阈值决策出是否存在闸片厚度到限的情况。
[0013]具体地，所述步骤二包括：
[0014]1)采用20层残差编码结构进行闸片分割，所述20层残差编码结构的通道数为[64,256,512,512]；
[0015]2)采用联合分割损失函数进行模型训练，所述联合分割损失函数为Loss＝iouloss+diceloss+focalloss，所述iouloss负责整块目标区域的监督，所述diceloss负责边缘分割的监督，所述focalloss负责像素级的分割监督；
[0016]3)保存训练好的模型。
[0017]具体地，所述亚像素边缘提取算法包括如下步骤：
[0018]步骤一：边缘检测滤波算子进行边缘检测
[0019]1)将所述二维闸片图像数据像素点边缘坐标进行灰度化处理；
[0020]2)通过高斯平滑滤波器对所述二维闸片图像数据像素点边缘坐标进行图像去噪；
[0021]3)计算位于
±
45
°
、
±
90
°
及
±
135
°
方向的二维闸片图像数据边缘像素点坐标的梯度幅值和方向；
[0022]4)通过非极大值抑制算法，去除非边界上的点；
[0023]5)通过滞后阈值算法，确定真正的边界；
[0024]步骤二：通过三次样条插值运算，将步骤一中得到的边缘坐标进行细分
[0025]步骤三：将步骤二中细分得到的边缘坐标进行边缘细化，即可得到闸片边缘坐标的亚像素边缘坐标。
[0026]本专利技术的有益效果：
[0027]1)通过非接触式光学成像方法获取到闸片图像，然后利用深度学习技术，通过大量的样本训练出一个闸片分割网络，然后利用分割结果，结合标定信息计算出闸片的厚度，通过亚像素提取，提高了计算精度；
[0028]2)适用于非接触式测量，对被检测目标无损伤；
[0029]3)基于人工智能算法，算法适应性好，可以适应不同的光照、不同的闸片类型，检测准确率高，便于推广应用。
附图说明
[0030]图1本专利技术检测流程示意图；
具体实施方式
[0031]下面结合具体实施方式对本专利技术进行详细的说明。
[0032]本专利技术基于传统算法的弊端，通过深度神经网络提取特征，直接实现端到端的检测，能够实现高速度、高精度的闸片定位分类与分割，并通过标定得到闸片真实物理厚度值，具有快速准确复用性强的特点。
[0033]如图1检测流程示意图所示，本专利技术具体包括如下步骤：
[0034]步骤一：根据检测需求，通过AI目标检测网络定义闸片磨耗体区域，并对闸片进行数据标注；
[0035]步骤二：设计AI语义分割网络，利用闸片标注数据进行模型训练，并保存训练好的模型；
[0036]AI语义分割网络针对闸片分割任务具有特定的检测结构与损失函数。
[0037]该语义分割网络在经典网络unet的基础上做出优化，unet网络编码阶段采用conv+bn+relu的结构，该结构直接使用到闸片分割中效果不佳，无法达到分割精度要求，基于此设计了20层的残差编码结构，如下表所示：
[0038][0039]随着网络层数的加深，感受野的扩大，网络会学习到更多的语义信息，丢失部分的边缘位置信息，考虑到该点，对该残差结构的通道数也做出了调整，由原始[64,128,256,512]调整为[64,256,512,512]。通过增加中间网络对边缘信息的关注达到提升分割效果的目的。
[0040]训练阶段，为了让网络更加关注到边缘信息的提取，设计了联合的分割损失函数。Loss＝iouloss+diceloss+focalloss
[0041]其中iouloss负责整块目标区域的监督，diceloss负责边缘分割的监督，focalloss负责像素级的分割监督。通过该联合损失函数进一步优化整体分割效果。
[0042]步骤三：设备采集到的二维闸片图像数据通过无线或者有线的方式，传输至AI闸片语义分割网络，AI闸片语义分割网络输出闸片的像素点边缘坐标及闸片类型；
[0043]步骤四：利用亚像素边缘提取算法，得到二维闸片图像数据像素点边缘坐标的亚像素边缘坐标；
[0044]步骤五：将二维闸片图像数据的亚像素边缘坐标输入标定计算模块；标定计算模块分离出二维闸片图像数据亚像素边缘坐标中左右边缘匹配的坐标点，组成直线段；标定计算模本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于深度学习技术的机车车辆闸片磨耗检测方法，其特征在于：具体包括如下步骤：步骤一：根据检测需求，通过AI目标检测网络定义闸片磨耗体区域，并对闸片进行数据标注；步骤二：设计AI语义分割网络，利用所述闸片标注数据进行模型训练，并保存训练好的模型；步骤三：设备采集到的二维闸片图像数据通过无线或者有线的方式，传输至AI闸片语义分割网络，所述AI闸片语义分割网络输出闸片的像素点边缘坐标及闸片类型；步骤四：利用亚像素边缘提取算法，得到所述二维闸片图像数据像素点边缘坐标的亚像素边缘坐标；步骤五：将所述二维闸片图像数据的亚像素边缘坐标输入标定计算模块；所述标定计算模块分离出所述二维闸片图像数据亚像素边缘坐标中左右边缘匹配的坐标点，组成直线段；所述标定计算模块通过已经标定好的参数，将所述直线段的像素长度转换成物理长度；最后基于概率统计的异常值剔除算法，剔除所述直线段中的异常直线段，得到所有保留直线段的长度平均值即为闸片厚度；步骤六：将所述闸片厚度输入闸片厚度超限报警模块，根据报警阈值决策出是否存在闸片厚度到限的情况。2.根据权利要求1所述的一种基于深度学习技术的机车车辆闸片磨耗检测方法，其特征在于：所述步骤二具体包括：1)采用20层残差编码结构进行闸片分割，所述20层残差编码结构的通道数为[...

【专利技术属性】
技术研发人员：胡平，史时喜，赵留辉，侯小祥，严飞，周航博，
申请(专利权)人：中铁第一勘察设计院集团有限公司，
类型：发明
国别省市：