一种机车轮对踏面剥离识别方法技术

本发明专利技术公开了一种机车轮对踏面剥离识别方法，包括采集踏面剥离图像构建正样本图像样本集，采集未踏面剥离图像样本构建负样本图像样本集，对正样本图像样本集和负样本图像样本集进行人工标注获得真实框；利用聚类算法对真实框进行聚类处理，获得特征值；利用yolo神经网络，根据特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框自动筛选先验框，将含有剥离情况的先验框进行分类和回归预测，得到分类损失和位置损失；利用误差反向传播算法离线训练分类损失和位置损失进而优化神经网络中的权重参数。本发明专利技术基于yolo神经网络进行训练和检测，检测速度快，准确率高，仅通过图片和录像就能进行检测，极大的节约了资源和时间成本。资源和时间成本。资源和时间成本。

【技术实现步骤摘要】
一种机车轮对踏面剥离识别方法


[0001]本专利技术涉及机车识别领域，尤其涉及一种机车轮对踏面剥离识别方法。

技术介绍

[0002]现有的机车轮对踏面剥离识别方法通常利用光电技术非接触测量踏面缺陷的方法。用精密的激光位移传感器对踏面进行快速扫描，采集值经过处理后转换为数字图像，再通过传统的信号处理技术识别表面是否存在缺陷。这种识别方法需要精密激光位移传感器，造价高，应用场合限制。需要用到传统信号处理方法，会受到环境干扰，需要人工干预检测，识别速度慢。

技术实现思路

[0003]本专利技术提供一种机车轮对踏面剥离识别方法，以克服现有的剥离识别方法识别速度慢等技术问题。
[0004]为了实现上述目的，本专利技术的技术方案是：
[0005]一种机车轮对踏面剥离识别方法，包括以下步骤：
[0006]步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本，保留满足设定分辨率的图像样本，构成正样本图像样本集；
[0007]步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本，保留满足设定分辨率的图像样本，构成负样本图像样本集；
[0008]步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框；
[0009]步骤4、利用聚类算法对真实框进行聚类处理，获得特征值；
[0010]步骤5、利用yolo神经网络，根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框；
[0011]步骤6、通过yolo神经网络自动筛选先验框，将含有剥离情况的先验框进行分类和回归预测，得到分类损失和位置损失；
[0012]步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数；
[0013]步骤8、将权重参数保存为文本格式，通过分类器预测出剥离情况。
[0014]进一步的，所述步骤6中得到位置损失具体方法为：通过逻辑回归预测出先验框与真实框的偏移量，再通过交并补运算得到真实框与先验框之间的IOU值，即位置损失。
[0015]进一步的，计算IOU值的公式为：
[0016][0017]其中，J(A,B)为真实框与先验框之间的IOU值，|A∩B|为真实框的IOU值，|A∪B|为先验框的IOU值。
[0018]进一步的，步骤4中具体为：利用聚类算法对真实框进行聚类处理，使真实框划分
为不同网格，每个网格生成不同规格的先验框。
[0019]进一步的，步骤5中通过yolo神经网络卷积对正样本图像样本集和负样本图像样本集进行不同倍数的卷积采样，得到不同尺寸的先验框。
[0020]有益效果：本专利技术仅基于yolo模型进行训练和检测，检测速度快，准确率高，仅通过固定的摄像头或录像机就能进行检测，不需要精密的光学仪器，避免了环境中的噪声、热干扰、电磁波干扰等环境因素对预测结果产生的影响，同时摄像头和录像机能够自动捕捉发生剥离的区域，不需要人工干预，极大的节约了资源和时间成本。
附图说明
[0021]为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍，显而易见地，下面描述中的附图是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其他的附图。
[0022]图1为本专利技术机车轮对踏面剥离识别方法流程图。
具体实施方式
[0023]为使本专利技术实施例的目的、技术方案和优点更加清楚，下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0024]本实施例提供了一种机车轮对踏面剥离识别方法，如图1，包括以下步骤：
[0025]一种机车轮对踏面剥离识别方法，其特征在于，包括以下步骤：
[0026]步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本，保留满足设定分辨率的图像样本，构成正样本图像样本集；
[0027]步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本，保留满足设定分辨率的图像样本，构成负样本图像样本集；
[0028]步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框；
[0029]步骤4、利用聚类算法对真实框进行聚类处理，获得不同的大小和尺寸值，即特征值；
[0030]步骤5、利用yolo神经网络，根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框；
[0031]步骤6、通过yolo神经网络自动筛选先验框，将含有剥离情况的先验框进行分类和回归预测，得到分类损失和位置损失；
[0032]步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数；
[0033]步骤8、将权重参数保存为文本格式，通过分类器预测出剥离情况。
[0034]在具体实施例中，所述步骤6中得到位置损失具体方法为：通过逻辑回归预测出先验框与真实框的偏移量，再通过交并补运算得到真实框与先验框之间的IOU值，即位置损失。
[0035]在具体实施例中，计算IOU值的公式为：
[0036][0037]其中，J(A,B)为真实框与先验框之间的IOU值，|A∩B|为真实框的IOU值，|A∪B|为先验框的IOU值。
[0038]在具体实施例中，步骤4中具体为：利用聚类算法对真实框进行聚类处理，使真实框划分为不同网格，每个网格生成不同规格的先验框。首先将得到的真实框划分为13*13个网格，每个网格会生成长宽不同，大小不同的9个先验框。先验框的大小由K
‑
Means聚类算法对所有预测框的长宽聚类得到；每个先验框都会预测三类值：1.位置信息包括中心坐标tx和ty，框的宽和长；2.有无目标预测；3.所属正负样本的剥离情况；
[0039]在具体实施例中，步骤5中通过yolo神经网络卷积对正样本图像样本集和负样本图像样本集进行不同倍数的卷积采样，得到不同尺寸的先验框；具体的，步骤5中通过多层卷积神经网络得到32倍，16倍，8倍三次不同大小的采样，获得不同尺寸先验框，每个先验框所对应的感受野不同，这样对所获得的不规整数据集做目标检测十分有利，可以确定数据中不同大小的目标种类。
[0040]在实际剥离过程中，只有第一次采集的正样本图像样本集和负样本图像样本集需要人工标注，由于人工标注已经训练好yolo神经网络，所以后续再进行预测无需人工标注，采集正样本图像样本集和负样本图像样本集后即可识别剥离情况。
[0041]项目YOLO
‑
v3准确率91.5％精确率89.52％召回率94％假正率11％真正率94％F
‑
measure91.69％
[0042]表1：剥离识别准本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种机车轮对踏面剥离识别方法，其特征在于，包括以下步骤：步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本，保留满足设定分辨率的图像样本，构成正样本图像样本集；步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本，保留满足设定分辨率的图像样本，构成负样本图像样本集；步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框；步骤4、利用聚类算法对真实框进行聚类处理，获得特征值；步骤5、利用yolo神经网络，根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框；步骤6、通过yolo神经网络自动筛选先验框，将含有剥离情况的先验框进行分类和回归预测，得到分类损失和位置损失；步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数；步骤8、将权重参数保存为文本格式，通过分类器预测出剥离情况。...

【专利技术属性】
技术研发人员：王乃福，赵中林，高宏洋，
申请(专利权)人：中车大连电力牵引研发中心有限公司，
类型：发明
国别省市：