一种基于改进的Sobel算法钢轨表面损伤语义分割方法

曾树华 黄银秀

摘 要：为探寻钢轨表面伤损语义分割的有效方法，实现钢轨表面伤损准确检测。在经典Sobel算法的基础上，增加倾斜方向的检测，改变阀值；利用采集的带扎伤钢轨图片，在MATLAB2016版本进行对比检测实验；本方法成功检出两处伤损，无误检错检；结论：本方法检测效果在检出率和误检率都优于传统方法，有一定价值。

关键词：Sobel算法；钢轨表面损伤；语义分割；边缘检测

*基金：湖南省教育厅资助科研项目（19C1214）

0 引言

随着铁路运输的重载化、高频化愈加频繁，钢轨受到频繁强力撞击，伤损频繁。铁路运输间歇时间短暂，效率低下的人工检查方式致使作为铁路安全运营之基的钢轨无法及时有效排除隐患，给铁路安全运行带来巨大挑战。在钢轨损伤中，尤以表面伤损数量居多，检测效率低，最易被忽视，增加行车危险系数和维护成本。

近年来，高效自动的钢轨表面伤损检测技术成为学者们研究的重点，包括涡流检测技术、漏磁检测技术、超声检测技术和机器视觉技术等。其中，涡流检测方法在邻近多裂纹伤损情况下容易出现漏检错检；漏磁检测方法因其依赖人工目检，检测速度和准确度无法保证；超声检测方法存在近表面检测盲区，难以可靠检测表面伤损。故近年来，随着计算速度提升和先进算法的出现，机器视觉技术在钢轨表面伤损的应用越来越受到广大学者们的关注，基于机器视觉技术的钢轨表面伤损检测技术已经作为轨检辅助措施之一逐步开发出各种轨 检车[1]。

边缘检测算法中的微分算子法在其他机器视觉应用领域得到广泛应用，其中Canny和Log算子的检测精度较高，受到广泛关注，但因其算法复杂，而不适合实时性高的领域；Roberts和Prewitt算子的计算工作量小，但精度不高，在一些精度要求不高的場景有较广应用；Sobel算子法的算法更简单，特别适合实时性要求高的钢轨检测领域[2]，为此，各种提升检测精度的改进Sobel算子法陆续推出。刘源改进了Sobel算子检测方法，利用四方向结合动态阈值得到更好的边缘检测效果[3]。张雯柏结合二值形态处理将钢轨轮廓从背景中检测出 来[4]。蒋超等人通过计算像素的一阶导数的局部最大值和二阶导数的过零点来连接适当的像素，以获得缺陷边缘，该方法虽然具有有效的噪声抑制，但耗时较长[5]。潘少伟等人利用八个方向的Sobel算子对岩心图像边缘进行检测，取得不错的效果[6]，本文拟引入对钢轨表面伤损进行检测。

1 经典Sobel算子

经典Sobel算子自提出以来被广泛使用于各种边缘检测，该算子的思路是先使用3×3的卷积模板对被检图进行加权平均，以便降低边缘模糊程度，其卷积模板如图1所示。

具体算法步骤如下：

1）在水平方向和竖直方向用Sobel算子卷积模板遍历全部图像数据，图像数据点为模板的中心点。

2）图像数据中每个像素点对应的两个模板进行离散卷积运算，则每个像素点对应有两个值，取两个值中的较大者为像素点的灰度值，得到图像对应的新数据集。

3）边缘判断：规定一阈值于新数据集进行比较，若数据集中数据大于阀值判断为边缘点，连接所有边缘点得到图像边缘。

2 改进的Sobel算子检测方法

改进的Sobel算子的像素点梯度利用5×5梯度模板进行计算，将检测角度增加到8个，图2是其梯度模版矩阵分别定义。改进的Sobel算子增加了斜线方向的权重，有利于更细节的特征信号提取。在此基础上，调整了判断阀值。

3 实验结果与分析

为了验证改进的算法对钢轨表面伤损定位的准确性，对采集的钢轨表面伤损图片进行比对伤损识别实验[6]。采用MATLAB2016软件进行编程实验，结果如图3所示。其中图3（a）为带扎伤钢轨原图，图3（b）为传统Sobel算子检测图，采用的是MATLAB自带的Sobel算子，图3（c）是改进的Sobel算子检测效果图。由图3可知，传统Sobel算子在检测钢轨表面扎伤伤损时存在漏检和误检，效果差，改进的Sobel算子检测方法成功的检测出两处扎伤，无误检漏检。

4 结束语

本文以传统Sobel算子为基础，改进了Sobel算子边缘检测方法：将其检测方向增加到8个，并更改了判断阀值，并应用于钢轨表面扎伤检测实验中。新算法经实验表明：能有效检测钢轨表面，无错检漏检。本方法实验数量有限，效果尚需进一步大数据验证。

参考文献：

[1] 田贵云，高斌，高运来，等.铁路钢轨缺陷伤损巡检与监测技术综述[J].仪器仪表学报，2016，37（8）：1763-1780.

[2] 管宏蕊，丁辉.图像边缘检测经典算法研究综述[J].首都师范大学学报（自然科学版），2009，30（S1）：66-69.

[3] 刘源，夏春蕾.一种基于Sobel算子的带钢表面缺陷图像边缘检测算法[J].电子测量技术，2021，44（3）：138-143.

[4] 张雯柏，柴晓冬，郑树彬，等.基于二值形态学算子的轨道图像分割新算法[J].测控技术，2018，37（10）：10-13，21.

[5] 蒋超，牛宏侠.基于改进Radon变换的直线钢轨识别算法[J].铁道标准设计，2017，61（4）：19-22.

[6] 王平，刘泽，王嵬，等.基于数字图像处理和特征提取的钢轨表面缺陷识别方法[J].现代科学仪器，2012（2）：24-28.

[7] 甘金瑞，李启元等.基于分层提取器的可视轨道表面检查系统[J]. IEEE传感器学报，2017，17（23）：7935-7944.

3993501908201