一种基于深度学习和结构信息的高铁场景感知方法技术

本发明专利技术公开了一种基于深度学习和结构信息的高铁场景感知方法，包括以下：步骤1：获取轨道图像，分为训练集和测试集，对训练集中的图像进行标注形成数据集；步骤2：构建SSD网络模型，并构造损失函数；步骤3：采用步骤1形成的数据集，对步骤2得到的网络进行迭代训练得到训练模型；步骤4：对需要检测感知的视频按帧输入到步骤3得到的训练模型中，提取特征，得到扣件和挡肩的位置和类别信息，根据扣件和挡肩的位置和类别信息区分道岔和普通轨道；步骤5：分别对步骤4中定位结果的位置信息进行聚类，完成钢轨和轨枕的感知；本发明专利技术可对道岔区域的轨道部件进行检测和语义分割，检测精度高，检测速度快。

【技术实现步骤摘要】
一种基于深度学习和结构信息的高铁场景感知方法
本专利技术涉及基于图像处理的轨道部件检测方法，具体涉及一种基于深度学习和结构信息的高铁场景感知方法。
技术介绍
铁路交通运输作为社会经济发展的重要支撑、人民生活中不可或缺的交通工具，在整个社会发展中具有举足轻重的作用；特别是针对我国幅员辽阔、人口流动量大、资源分布不均衡等因素；铁路运输以其运能大、运输成本低、占地面积相对较少等优点，在各类公共交通工具中占据绝对的优势；随着中国铁建设的推进，国家计划到2020年，中国高速铁路营运里程将达到3万公里，高速铁路网覆盖80％以上的大城市。铁轨设备的维护与检测是铁路交通技术中备受关注的问题，也是现阶段投入研发资金最多的项目之一；其中，高铁轨道主要由扣件、钢轨、轨枕、轨道板等构成，传统检测方法由经验丰富的巡道工完成，工人定期沿线路巡查，发现并报告异常部件；人工检测的劳动强度很大、工作环境非常恶劣，线上行驶车辆对检测员工的人身安全也构成潜在威胁；传统的铁路巡检方法已难以满足如今高速铁路运营的发展需要。随着机器视觉的发展，基于图像的线路视觉检测技术受到越来越多的重视，但传统的视觉检测方法通常只检测某一单独本文档来自技高网...

【技术保护点】
1.一种基于深度学习和结构信息的高铁场景感知方法，其特征在于，包括以下：步骤1：获取轨道图像，分为训练集和测试集，对训练集中的图像进行标注形成数据集；步骤2：构建SSD网络模型，并构造损失函数；步骤3：采用步骤1形成的数据集，对步骤2得到的网络进行迭代训练得到训练模型；步骤4：对需要检测感知的视频按帧输入到步骤3得到的训练模型中，提取特征，得到扣件和挡肩的位置和类别信息，根据扣件和挡肩的位置和类别信息区分道岔轨道和普通轨道；步骤5：分别对步骤4中定位结果的位置信息进行聚类，完成钢轨和轨枕的感知。

【技术特征摘要】
1.一种基于深度学习和结构信息的高铁场景感知方法，其特征在于，包括以下：步骤1：获取轨道图像，分为训练集和测试集，对训练集中的图像进行标注形成数据集；步骤2：构建SSD网络模型，并构造损失函数；步骤3：采用步骤1形成的数据集，对步骤2得到的网络进行迭代训练得到训练模型；步骤4：对需要检测感知的视频按帧输入到步骤3得到的训练模型中，提取特征，得到扣件和挡肩的位置和类别信息，根据扣件和挡肩的位置和类别信息区分道岔轨道和普通轨道；步骤5：分别对步骤4中定位结果的位置信息进行聚类，完成钢轨和轨枕的感知。2.根据权利要求1所述的一种基于深度学习和结构信息的高铁场景感知方法，其特征在于，所述步骤4中提取特征，得到扣件和挡肩的位置和类别信息的过程如下：在输入训练模型的每一帧图像中随机选取整幅图像的多个区域；通过卷积获取每个区域的评分；若评分值大于设定阈值则判断为扣件和挡肩；若评分值小于设定阈值则丢弃该结果，遍历所有区域，则将扣件和挡肩标注完成，得到位置和类别信息。3.根据权利要求1所述的一种基于深度学习和结构信息的高铁场景感知方法，其特征在于，所述步骤5中通过DBSCAN算法进行聚类。4.根据权利要求3所述的一种基于深度学习和结构信息的高铁场景感知方法，其特征在于，所述聚类过程如下：S1：设置步骤4检测到的目标的x方向坐标位置D，半径eps和密度阈值Minpts；S2：任意选定一个目标p；S3：判断其是否已经属于某个簇或已经成为噪声点，若是则返回步骤S2，若否则转入步骤S4；S4：判断p的邻域范围的点是否小于Minpts，若是则标记p为噪声点，转入步骤S2；若否则转入步骤S5；S5：遍历p邻域半径范围内未被标记的点q；S6：判断q的邻域范围的点是否小于Minpts，若是则转入步骤S5，若否则转入步骤S7；S7：将q邻域内未归入其他簇的点加入到集合C中，结束。5.根据权利要求1所述的一种基于深度学习和结构信息的高铁场景...

【专利技术属性】
技术研发人员：李兆洋，罗建桥，李柏林，程洋，
申请(专利权)人：西南交通大学，
类型：发明
国别省市：四川,51