[发明专利]一种铁路扣件缺陷检测方法和装置

说明书

本发明提供了一种铁路扣件缺陷检测方法和装置，涉及铁路检测技术领域。所述铁路扣件缺陷检测方法包括：根据采集的铁路扣件图像信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本；根据目标算法对所述第一目标样本进行扩充，确定扩充后的第二目标样本；根据所述第二目标样本和所述第一目标样本，生成用于检测铁路扣件的检测模型；将所述待检测样本输入所述检测模型，确定检测结果。本发明的方案根据扩充后的第二目标样本以及具有位置信息的第一目标样本，生成用于检测铁路扣件的检测模型，解决了现有技术中缺陷样本不平衡的问题，提高了检测准确率。

技术领域

本发明涉及铁路检测技术领域，特别涉及一种铁路扣件缺陷检测方法和装置。

背景技术

随着铁路网络布局的推进和高铁技术的快速发展，铁路运营维护检测承担着越来越重要的责任。列车的安全运行需要铁路系统的安全保障，而铁路扣件是连接铁轨和枕木，保证轨道距离的重要组件，在高速铁路检测系统中起着重要的作用。然而，由于铁路基础设施分布跨度大、环境变化大，以及列车振动等的影响，铁路扣件会出现松脱、断裂，甚至缺失等缺陷，这会严重影响铁路运行安全，甚至会引发铁路脱轨事故。因此，铁路扣件需要定期进行检测，以保证铁路的安全运行。

目前，我国铁路线路的扣件类型各有差异，传统的检测方法需要依靠人工操作，该方法耗费大量时间、劳动强度高、效率低，同时会给检查人员带来未知的安全隐患。随着高速铁路的快速发展，铁路扣件的检测面临着大的挑战，因此，研究快速准确的铁路扣件识别算法具有很大的需求。

现有技术中基于二维图像的铁路扣件缺陷检测方法发展迅速，结合深度学习算法加快了扣件的检测速度和检测精度，并且已经发展了相应的检测系统。但是由于二维图像受光照方向和光线条件影响大，使得图像质量较低的铁路铁路扣件缺陷检测较为困难，召回率较低。除此之外，由于铁路扣件的样本数量庞大且缺陷样本的数量远少于正常样本的数量，使得检测模型不能有效记录和提取缺陷样本的特征，尤其在利用深度学习算法模型检测时，模型更多的学习到正常样本的特征属性，使得误检误报率增加，影响最后的检测结果。

现有技术中利用三维信息进行扣件检测的研究较少且不成熟，并且多是利用模板匹配进行扣件分割和定位，速度较慢，针对缺陷样本不均衡的问题没有有效解决方法。

综上所述，基于机器视觉的二维图像铁路扣件缺陷检测方法中的弊端是：铁路扣件缺陷样本少，样本标注困难以及二维铁路扣件缺陷检测受光照条件影响大；基于三维信息进行扣件检测的方法不成熟，针对缺陷样本不均衡的问题没有有效解决。

发明内容

本发明实施例提供一种铁路扣件缺陷检测方法和装置，以解决现有技术的检测方法中针对样本不均衡和样本标注效率低的问题。

为了解决上述技术问题，本发明采用如下技术方案：

本发明实施例提供一种铁路扣件缺陷检测方法，包括：

根据采集的铁路扣件图像信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本；

根据目标算法对所述第一目标样本进行扩充，确定扩充后的第二目标样本；

根据所述第二目标样本和所述第一目标样本，生成用于检测铁路扣件的检测模型；

将所述待检测样本输入所述检测模型，确定检测结果。

可选的，根据采集的铁路扣件图像信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本，包括：

获取铁路扣件图像信息，所述铁路扣件图像信息包括深度图像和强度图像；所述深度图像和强度图像的像素一一对应；

根据所述深度图像的像素，确定所述铁路扣件图像信息中铁路扣件的位置信息；

根据所述铁路扣件的位置信息和所述强度图像的像素，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本。