铁路货车制动梁安全链脱落故障图像识别方法及系统

技术领域

本发明涉及货车制动梁安全链脱落故障图像识别方法及系统。

背景技术

列车制动系统是列车的重要组成部分，其性能的好坏和制动能力的大小直接涉及列车能否安全运行。制动梁是铁道车辆上最重要的部件之一，它的损伤故障对行车安全影响极大。在列车运行过程中，制动梁承受较大的交变载荷及冲击力，制动时承受制动力以及车轮对闸瓦的反作用力，受力状况较为恶劣。制动梁安全链与梁体的连接是由一个整体的安全链卡子和一条眼环螺栓紧固在一起。当制动梁安全链发生脱落故障，这不但会失去对制动梁体的固定作用，也会由于安全链与梁体直接摩擦造成梁体局部磨耗，给货车的运行带来安全隐患。现采用的人工逐张看图的检车作业方式，存在受人员素质、责任心影响，错漏检问题时有发生，作业质量难以保证，以及人工成本巨大、效率低下等问题。

所以针对制动梁安全链脱落故障的自动化检测具有重要意义。通过结合图像处理和深度学习技术，实现故障自动识别和报警，有效提高检车作业质量和效率。

发明内容

本发明的目的是为了解决现有目标检测方法中铁路货车制动梁安全链位于制动梁后方，存在较为明显的遮挡和重叠，导致检测精度低的问题，而提出铁路货车制动梁安全链脱落故障图像识别方法及系统。

铁路货车制动梁安全链脱落故障图像识别方法具体过程为：

建立原始样本图像数据集；

所述图像为铁路货车制动梁安全链图像；

基于原始样本图像数据集对网络模型进行训练，直至模型收敛，获得训练好的网络模型；具体过程为：

网络结构为SSD网络；

训练过程采用改进的Repulsion Loss损失函数以及DIoU-NMS非极大值抑制方法去除重复框；

改进的Repulsion Loss损失函数具体为：

Repulsion loss由三个损失项构成；

L＝L

第一项L

第二项L

第三项L

α、β表示权重；

对第二项中周围真实目标框进行改进，改进后的周围真实目标框选取方式用下式表示：

其中D表示默认框，

基于获得的训练好的网络模型对待测图像进行故障识别。

优选地，建立原始样本图像数据集；具体过程为：

一、获取铁路货车线阵图像；

二、基于铁路货车线阵图像，获取铁路货车制动梁安全链部件粗定位图像；

三、对铁路货车制动梁安全链部件粗定位图像进行数据集扩增；

四、基于扩增后数据集，建立原始样本图像数据集。

优选地，一中获取铁路货车线阵图像；具体过程为：

分别在货车轨道底部搭建高清设备，对通过的货车进行拍摄，获取货车底部的图像。

优选地，二中基于铁路货车线阵图像，获取铁路货车制动梁安全链部件粗定位图像；具体过程为：

根据货车轴距信息、转向架类型信息对货车底部的图像中的制动梁安全链位置进行粗定位，获取部件粗定位图像。

优选地，三中对铁路货车制动梁安全链部件粗定位图像进行数据集扩增；具体过程为：

扩增形式包括图像的旋转、平移、缩放、镜像操作，每种操作都是在随机条件下进行的。

优选地，四中基于扩增后数据集，建立原始样本图像数据集；具体过程为：

原始样本图像数据集包括：部件粗定位图像集与标记信息集；

部件粗定位图像集与标记信息数据集之间是一一对应的，即每个部件粗定位图像对应一个标记数据。

优选地，DIoU-NMS非极大值抑制方法具体过程为：

假设SSD网络模型的一个默认框集为B，默认框集B对应的类别置信度得分集为s，对于得分最高的预测框M，将第i个默认框的分类得分进行更新：

其中，s

优选地，基于获得的训练好的网络模型对待测图像进行故障识别；具体过程为：

步骤六一：部件粗定位：

根据转向架类型信息及部件所在区域的先验知识，取出包含部件的子区域图像；

步骤六二：故障判定：

将子区域图像输入训练好的模型，完成子区域图像中的制动梁安全链部件检测，判定是否存在故障，若是，执行步骤六三；若否，继续检测下一张图像；

步骤六三：上传报警：

根据故障信息，将故障位置和故障类别生成报文，并上传至报警平台。

优选地，权重α、β取值均为0.5。

基于铁路货车制动梁安全链脱落故障图像识别系统用于执行基于深度学习的铁路货车制动梁安全链脱落故障图像识别方法。

本发明的有益效果：

与其他货车部件的目标检测相比，制动梁安全链位于制动梁后方，存在较为明显的遮挡。其中遮挡存在两种情况：一种是由于制动梁的遮挡导致安全链显示不完整的情况，这会使得深度学习检测模型学习不到完整的特征进而导致误检；另一种是由于链条的柔性特征，这会使得不同节的链条堆叠在一起，检测模型无法分辨而导致漏检。本专利从损失函数的角度出发，采用一种改进的Repulsion Loss损失函数以及DIoU-NMS非极大值抑制方法去除重复框，解决由于遮挡、重叠情况引起的制动梁安全链检测精度低的问题。

本发明基于与默认框匹配的真实目标框进行选取周围真实目标框。以真实目标框为标准，计算它与周围的其他真实目标框的IoU，具有最大IoU的选为周围真实目标框，这种不依靠预测框进行选取的策略，即使出现了预测框偏差很大的情况，也不会影响训练收敛，使得算法更加鲁棒。

本发明采用DIoU-NMS非极大值抑制方法，在抑制准则中不仅对重叠区域进行分析，而且还计算两个检测框之间的中心点距离，有效地避免了遮挡和重叠情况下目标检测精度低的问题，本发明可以有效提升遮挡、重叠情况下目标检测的精度。

本发明利用在货车轨道底部的高清成像设备，对高速运动的货车进行拍摄，获取货车底部的高清图像。根据轴距信息和转向架类型等先验知识，获取包含部件的粗定位图像。对图像进行收集、整理以及数据扩增获取训练图像样本集。根据故障类型搭建合适的深度神经网络，并进行多次训练直至模型收敛，获得相应参数。在识别阶段，加载参数，并将拍摄到的图像输入网络得到预测结果，根据预测结果判断是否为故障，如果是故障则对故障区域进行报警。

本发明将自动识别技术引入货车故障检测，实现故障自动识别及报警，人工只需对报警结果进行确认，有效节约人力成本，提高作业质量和作业效率。将深度学习算法应用到制动梁安全链脱落故障自动识别中，相较传统的机器视觉检测方法具有更高的准确性、稳定性。

附图说明

图1为本发明模型训练流程图；

图2为SSD网络结构图；

图3为故障判定流程图。

具体实施方式

需要特别说明的是，在不冲突的情况下，本申请公开的各个实施方式之间可以相互组合。

具体实施方式一：结合图1、图3说明本实施方式，本实施方式铁路货车连接拉杆折断故障识别方法具体过程为：

建立原始样本图像数据集；

所述图像为铁路货车制动梁安全链图像；

基于原始样本图像数据集对网络模型进行训练，直至模型收敛，获得训练好的网络模型；具体过程为：

网络结构为SSD网络；

训练过程采用改进的Repulsion Loss损失函数以及DIoU-NMS非极大值抑制方法去除重复框；

改进的Repulsion Loss损失函数具体为：

Repulsion loss由三个损失项构成；

L＝L

第一项L

第二项L

第三项L

α、β表示权重；

对第二项中周围真实目标框进行改进，改进后的周围真实目标框选取方式用下式表示：

其中D表示默认框，

基于获得的训练好的网络模型对待测图像进行故障识别。

在训练数据集建立完成后，选用深度学习目标检测算法对制动梁安全链部件进行检测，模型训练流程如图1所示。

与其他货车部件的目标检测相比，制动梁安全链位于制动梁后方，存在较为明显的遮挡。其中遮挡存在两种情况：一种是由于制动梁的遮挡导致安全链显示不完整的情况，这会使得深度学习检测模型学习不到完整的特征进而导致误检；另一种是由于链条的柔性特征，这会使得不同节的链条堆叠在一起，检测模型无法分辨而导致漏检。本专利从损失函数的角度出发，采用一种改进的Repulsion Loss损失函数以及DIoU-NMS非极大值抑制方法去除重复框，解决由于遮挡情况引起的制动梁安全链检测效果差的问题。

1.SSD目标检测算法

目前常用的目标检测算法包括Faster R-CNN和YOLO，其中Faster R-CNN摈弃了之前目标检测生成候选区域的方法，通过RPN网络生成候选区域进行检测，使得其检测精度很高，但是速度较慢。YOLO则直接通过回归得出目标位置并判定其类别，将检测任务转变成回归任务。少了候选区域的生成过程，直接一步就获得目标边框位置和类别预测，大大加快了检测速度，然而相对地检测精度又变差了。

本发明采用目标检测算法SSD，它综合了YOLO和Faste R-CNN的优点，使得算法在保持高精度的同时，速度也达到了实时检测的要求。

图2是SSD的网络结构，其主要包括三部分：基础网络、特征提取网络、检测网络；

(1)基础网络：原文中的基础网络是用VGG16搭建而成，作者将原来VGG16中的fc6，fc7全连接层分别用卷积核为3×3×1024和1×1×1024的卷积层代替，因为全连接的结构会破坏特征的位置信息；

(2)特征提取网络：将上述修改后的VGG16中的conv7_2卷积特征图后通过1×1，3×3的卷积组合分别生成conv8_2、conv9_2、conv10_2、conv11_2这4个特征图，和conv4_3、conv7特征图构成了SSD的特征金字塔，生成特征图的网络即为特征提取网络。一般的检测算法都是用一张特征图进行检测，SSD采用了6张特征图。

(3)检测网络：特征提取网络总共生成了6张特征图，检测网络对每一张特征图再分别采用两个并列的3×3卷积核进行卷积，生成一张每个特征点代表一个坐标值的坐标偏置图和一张每个特征点代表某个类别预测得分的类别置信度图。

2.Repulsion Loss设计原理

Repulsion Loss的解决思路是设计一种全新的损失函数，其包括两个部分：一部分是使得预测框更接近真实目标框的吸引项，另一部分是使得预测框尽可能远离周围其他真实目标框和其他预测框的排斥项。这种设置能使得预测框在训练当中能够逐渐向匹配的真实目标靠近，并且避免因为进入到错误的目标区域而被非极大抑制滤除。

3.Repulsion Loss函数定义

Repulsion loss由三个损失项构成；

L＝L

(1)Attr Loss

第一项L

其中P是默认框，P

(2)RepGT Loss

第二项L

其中

(3)RepBox Loss

第三项L

其中预测框

4.Repulsion Loss在SSD上的实现细节及其改进

与R-CNN系列算法中普遍使用的通过候选区域进行分类和回归的方法不同，SSD采用的是默认框的方式进行分类和回归，所以在Repulsion Loss的实现细节上也存在着很多不同。

Repulsion Loss相对于SSD原来的多任务损失函数主要增加了RepGT Loss和RepBox Loss两种损失。其中RepGT Loss的存在使得预测框尽量远离匹配目标附近的真实目标框，因为原文中Repulsion Loss的实现是基于默认框的，选取的是除了与默认框匹配的真实目标框以外IoU最大的周围真实目标框，具体选取方法如式所示。

其中P为默认框，G为真实目标框，

但是对于SSD来说，它没有默认框，取而代之的是默认框。而默认框是基于特征图选取的一系列位置固定的框，它如果和真实目标框匹配的话就会被选为正样本，然后基于这些正样本默认框进行分类和回归。但是这种方案也会存在一个问题，因为网络在训练的开始阶段，其权重是随机初始化的，所以预测框一开始可能会偏离得比较大，导致预测框下一步的定位难以预测，进而影响训练的收敛甚至检测性能。

为了解决这个问题，本专利提出的一个改进方案就是，基于默认框匹配的真实目标框进行选取周围真实目标框。以真实目标框为标准，计算它与周围的其他真实目标框的IoU，具有最大IoU的选为周围真实目标框，这种不依靠预测框进行选取的策略，即使出现了预测框偏差很大的情况，也不会影响训练收敛，使得算法更加鲁棒。改进后的周围真实目标框选取方式可以用下式表示。

其中D表示默认框，

对于RepBox Loss，它的作用是使预测不同目标的预测框相互远离。但是在实现的时候，如果两个预测框靠的太近的话，RepBox Loss的设计机制会导致两个预测框离开对方真实目标框的范围，这样就使得预测框定位产生偏差。为解决这个问题，我们通过设置一个重叠阈值，即允许预测框和周围目标真实框存在一定的交集，只有超过这个阈值的重叠才会计算RepBox Loss。

5.DIoU-NMS非极大值抑制方法

NMS是抑制非极大值元素的方法总称，用来去除相邻的重复检测框，留下最精准的检测框。在传统的NMS中，利用定义的IoU来抑制重复检测框，但由于是仅对重叠区域进行分析，因此当检测目标存在遮挡和重叠情况时，容易发生不符合预期的抑制效果，将原本需要检测的目标过滤掉，造成目标漏检的情况发生。为解决这个问题，本发明采用DIoU-NMS非极大值抑制方法，在抑制准则中不仅对重叠区域进行分析，而且还计算两个检测框之间的中心点距离，有效地避免了遮挡和重叠情况下目标漏检的问题。

假设模型的一个默认框集为B，其对应的类别置信度集为s，对于得分最高的预测框M，将分类得分进行更新：

其中，s

对于两个中心点较远的矩形框，有可能位于不同的目标上，因此不进行滤除。通过同时分析两个矩形框的IoU以及中心点之间的距离来进行检测框去除，可以有效提升遮挡、重叠情况下目标检测的精度。

具体实施方式二：本实施方式与具体实施方式一不同的是，建立原始样本图像数据集；具体过程为：

一、获取铁路货车线阵图像；

二、基于铁路货车线阵图像，获取铁路货车制动梁安全链部件粗定位图像；

三、对铁路货车制动梁安全链部件粗定位图像进行数据集扩增；

四、基于扩增后数据集，建立原始样本图像数据集。

其它步骤及参数与具体实施方式一相同。

具体实施方式三：本实施方式与具体实施方式一或二不同的是，一中获取铁路货车线阵图像；具体过程为：

分别在货车轨道底部搭建高清设备，对通过的货车进行拍摄，获取货车底部的图像。

采用线扫描，可实现图像无缝拼接，生成视野大、精度高的二维图像。

其它步骤及参数与具体实施方式一相同。

具体实施方式四：本实施方式与具体实施方式一至三之一不同的是，二中基于铁路货车线阵图像，获取铁路货车制动梁安全链部件粗定位图像；具体过程为：

根据货车轴距信息、转向架类型信息对货车底部的图像中的制动梁安全链位置进行粗定位，获取部件粗定位图像。

从底部大图中截取包含部件的局部区域图像，可有效减少故障识别所需时间、提升识别准确率。

其它步骤及参数与具体实施方式一至三之一相同。

具体实施方式五：本实施方式与具体实施方式一至四之一不同的是，三中对铁路货车制动梁安全链部件粗定位图像进行数据集扩增；具体过程为：

样本数据集的建立虽然包括各种条件下的图像，但为提高算法的稳定性，仍需要对样本数据集进行数据扩增。

扩增形式包括图像的旋转、平移、缩放、镜像操作，每种操作都是在随机条件下进行的，这样可以最大程度的保证样本的多样性和适用性。

由于货车部件可能受到雨水、泥渍、油渍、黑漆等自然条件或者人为条件的影响。并且，不同站点拍摄的图像可能存在差异。

在训练图像数据集的搜集的过程中，要保证多样性，尽量搜集其在各种条件、不同站点的图像。

其它步骤及参数与具体实施方式一至四之一相同。

具体实施方式六：本实施方式与具体实施方式一至五之一不同的是，四中基于扩增后数据集，建立原始样本图像数据集；具体过程为：

原始样本图像数据集包括：部件粗定位图像集与标记信息集；标记信息故障位置和故障类别；

原图像集为设备拍摄的包含制动梁安全链部件的粗定位图像；

标记信息集为包含部件的矩形子区域信息，通过人工标记的方式获取；

部件粗定位图像集与标记信息数据集之间是一一对应的，即每个部件粗定位图像对应一个标记数据。

其它步骤及参数与具体实施方式一至五之一相同。

具体实施方式七：本实施方式与具体实施方式一至六之一不同的是，DIoU-NMS非极大值抑制方法具体过程为：

假设SSD网络模型的一个默认框集为B，默认框集B对应的类别置信度得分集为s，对于得分最高的预测框M，将第i个默认框的分类得分进行更新：

其中，s

其它步骤及参数与具体实施方式一至六之一相同。

具体实施方式八：本实施方式与具体实施方式一至七之一不同的是，基于获得的训练好的网络模型对待测图像进行故障识别；具体过程为：

步骤六一：部件粗定位：

根据转向架类型信息及部件所在区域的先验知识，取出包含部件的子区域图像；

步骤六二：故障判定：

将子区域图像输入训练好的模型，完成子区域图像中的制动梁安全链部件检测，对部件位置和形态进行逻辑分析，判定是否存在故障，若是，执行步骤六三；若否，继续检测下一张图像；

步骤六三：上传报警：

根据故障信息，将故障位置和故障类别生成报文，并上传至报警平台。

其它步骤及参数与具体实施方式一至七之一相同。

具体实施方式九：本实施方式与具体实施方式一至八之一不同的是，权重α、β取值均为0.5。

其它步骤及参数与具体实施方式一至八之一相同。

具体实施方式十：本实施方式铁路货车制动梁安全链脱落故障图像识别系统用于执行基于深度学习的铁路货车连接拉杆折断故障识别方法。

本发明的上述算例仅为详细地说明本发明的计算模型和计算流程，而并非是对本发明的实施方式的限定。对于所属领域的普通技术人员来说，在上述说明的基础上还可以做出其它不同形式的变化或变动，这里无法对所有的实施方式予以穷举，凡是属于本发明的技术方案所引伸出的显而易见的变化或变动仍处于本发明的保护范围之列。