一种基于小样本目标检测的轨道交通检测方法技术

本发明专利技术公开了一种基于小样本目标检测的轨道交通检测方法，该方法包括以下步骤；S1、接收轨道交通巡检任务，获取轨道交通巡检区域；S2、获取轨道交通巡检区域内宏观上的测试图像及参考图像，并采用图像帧前后对比算法对测试图像及参考图像进行对比，实现对轨道交通宏观的筛选；S3、根据筛选结果进行自适应微观切换，完成变化区域的筛选；S4、采用小样本目标检测算法对变化区域的具体形态进行识别。本发明专利技术能够在轨道交通沿线检测是否有沉积，宏观是轨道旁边检测是否有车、人经过，微观是检测是否有石头、裂缝等，通过宏观与微观上的配合，准确的得到轨道交通变化区域及识别变化区域的具体形态。形态。形态。

【技术实现步骤摘要】
一种基于小样本目标检测的轨道交通检测方法


[0001]本专利技术涉及轨道交通
，具体来说，涉及一种基于小样本目标检测的轨道交通检测方法。

技术介绍

[0002]随着高速铁路、重载铁路、城际铁路、城市轨道交通等的高速发展，我国总体规模和建设速度都处于世界第一的位置。轨道交通的安全运行意义重大，因此轨道交通的检测方法的发展十分重要。轨道交通范围大、路程长，需要采用检测设备进行轨道交通检测。
[0003]专利号201910376394.3公开了一种轨道交通设备的巡检方法及装置，包括接收服务器发送的巡检任务单，所述巡检任务单包括目标检测设备集，获取采集装置采集的巡检图像信息，识别所述巡检图像信息中包括的实际检测设备集，对所述实际检测设备集与所述目标检测设备集进行集合运算，确定巡检结果，实现了对巡检任务的自动复核检查。
[0004]专利号201410100802.X公开了一种地铁隧道表面裂缝的检测方法，利用高速面阵相机对地铁隧道洞体表面进行高速图像采集；获得二值化图像；获得滤除不规则及分布无规律的噪声点的隧道表面图像；根据所述滤除不规则及分布无规律的噪声点的隧道表面图像进行隧道表面裂缝的检测，并在检测到裂缝后对其大小进行计算。
[0005]但是，以上专利在使用时存在以下问题：轨道检测的场景多样，宏观上轨道旁边有车、人经过，微观的可能有石头、裂缝等。且轨道交通的范围大、路程长，若针对微观区域进行目标检测，则效率低，没有做到宏观与微观之间的结合，无法做到效率的提升。同时若出现少量图像采集样本可用的情况下，很容易忽视新物体的细粒度的特征，会对结果产生很大误导。在小样本检测中，具有尺度感知修改的特征提取器倾向于过度拟合，无法进行泛化，导致基类和新类的性能下降。
[0006]针对相关技术中的问题，目前尚未提出有效的解决方案。

技术实现思路

[0007]针对相关技术中的问题，本专利技术提出一种基于小样本目标检测的轨道交通检测方法，以克服现有相关技术所存在的上述技术问题。
[0008]为此，本专利技术采用的具体技术方案如下：
[0009]一种基于小样本目标检测的轨道交通检测方法，该方法包括以下步骤；
[0010]S1、接收轨道交通巡检任务，获取轨道交通巡检区域；
[0011]S2、获取轨道交通巡检区域内宏观上的测试图像及参考图像，并采用图像帧前后对比算法对测试图像及参考图像进行对比，实现对轨道交通宏观的筛选；
[0012]S3、根据筛选结果进行自适应微观切换，完成变化区域的筛选；
[0013]S4、采用小样本目标检测算法对变化区域的具体形态进行识别。
[0014]进一步的，所述S2中采用图像帧前后对比算法对测试图像及参考图像进行对比，实现对铁轨宏观的筛选还包括以下步骤：
[0015]完成测试图像与参考图像对比的准备工作；
[0016]通过孪生神经网络分别对测试图像及参考图像的特征进行提取；
[0017]利用全卷积网络对提取的测试图像及参考图像的特征进行整合相加，并利用softmax进行分类，得到变化图像。
[0018]进一步的，所述完成测试图像与参考图像对比的准备工作还包括以下步骤：
[0019]设定测试图像与参考图像的大小为w
×
h
×
d；
[0020]设定变化的类别标签从集合l＝1，2
…
，N对齐到变化图I
w
×
h
；
[0021]其中，N是定义的语义变化的类别数，w表示图像的宽度、h表示图像的高度，d表示图像的通道数。
[0022]进一步的，所述孪生神经网络包括ResNet残差块、卷积层及反卷积层，所述卷积层的卷积核大小为1
×
1。
[0023]进一步的，所述采用图像帧前后对比算法对测试图像及参考图像进行对比之前若训练数据有限，则进行迁移学习，其中，所述迁移学习还包括以下步骤：
[0024]采用ResNet50作为预训练模型；
[0025]利用反卷积网络的concate将预训练模型的输出做和，并利用1
×
1的卷积核将维度减少至N；
[0026]使用softmax分类器进行分类，得到的变化图像的大小为w
×
h
×
N；
[0027]其中，w表示图像的宽度、h表示图像的高度，N是定义的语义变化的类别数。
[0028]进一步的，所述S4中采用小样本目标检测算法对变化区域的具体形态进行识别还包括以下步骤：
[0029]获取变化区域的图像和支持集，将变化区域的图像和支持集输入共享特征提取器，并生成查询和支持特征；
[0030]将查询和支持特征输入密集关系蒸馏模块，并经过一个3
×
3的卷积被编码为键和值；
[0031]在获取查询特征和支持特征的键/值图后，进行关系蒸馏。
[0032]进一步的，所述在获取查询特征和支持特征的键/值图后，进行关系蒸馏还包括以下步骤：
[0033]通过测量查询特征的关键映射和支持特征的关键映射之间的相似性得到支持特征的软权重，即以非局部方式执行像素级相似性，公式如下：
[0034]F(k
qi
,k
sj
)＝φ(k
qi
)
T
φ'(k
sj
)；
[0035]其中，i和j是查询和支持位置的索引，φ、φ'表示两个不同的线性变换，参数在训练过程中通过反向传播学习，形成一个动态学习的相似度函数，k
qi
,k
sj
是查询特征的关键映射和支持特征的关键映射，T为非零自然数。
[0036]进一步的，所述像素级相似性计算之后，通过试行softmax归一化以输出最终权重：
[0037][0038]其中，exp为以自然常数e为底的指数函数。
[0039]通过加权求和与产生的软权重来检索支持特征的值，并将支持特征的值与查询特征的值连接起来，输出公式为：
[0040]y＝concat[v
q
,W*v
s
]；
[0041]其中，*表示矩阵内积，W为最终权重，v
q
,v
s
是支持特征的值与查询特征的值，支持特征有N个，并带来了N个键值对，对N个输出结果进行求和获得最终结果，即细化的查询特征。
[0042]进一步的，所述将查询和支持特征输入密集关系蒸馏模块，并经过一个3
×
3的卷积被编码为键和值时，键用来度量查询特征和支持特征的相似度，值则存放具体的信息。
[0043]进一步的，所述获得最终结果之后，将细化的查询特征输入RPN网络，并输出区域建议；
[0044]以区域建议及特征为输入，通过RoIAlign输出尺度分别为4
×
4、8
×
8本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于小样本目标检测的轨道交通检测方法，其特征在于，该方法包括以下步骤；S1、接收轨道交通巡检任务，获取轨道交通巡检区域；S2、获取轨道交通巡检区域内宏观上的测试图像及参考图像，并采用图像帧前后对比算法对测试图像及参考图像进行对比，实现对轨道交通宏观的筛选；S3、根据筛选结果进行自适应微观切换，完成变化区域的筛选；S4、采用小样本目标检测算法对变化区域的具体形态进行识别。2.根据权利要求1所述的一种基于小样本目标检测的轨道交通检测方法，其特征在于，所述S2中采用图像帧前后对比算法对测试图像及参考图像进行对比，实现对铁轨宏观的筛选还包括以下步骤：完成测试图像与参考图像对比的准备工作；通过孪生神经网络分别对测试图像及参考图像的特征进行提取；利用全卷积网络对提取的测试图像及参考图像的特征进行整合相加，并利用softmax进行分类，得到变化图像。3.根据权利要求2所述的一种基于小样本目标检测的轨道交通检测方法，其特征在于，所述完成测试图像与参考图像对比的准备工作还包括以下步骤：设定测试图像与参考图像的大小为w
×
h
×
d；设定变化的类别标签从集合l＝1，2
…
，N对齐到变化图I
w
×
h
；其中，N是定义的语义变化的类别数，w表示图像的宽度、h表示图像的高度，d表示图像的通道数。4.根据权利要求3所述的一种基于小样本目标检测的轨道交通检测方法，其特征在于，所述孪生神经网络包括ResNet残差块、卷积层及反卷积层，所述卷积层的卷积核大小为1
×
1。5.根据权利要求4所述的一种基于小样本目标检测的轨道交通检测方法，其特征在于，所述采用图像帧前后对比算法对测试图像及参考图像进行对比之前若训练数据有限，则进行迁移学习，其中，所述迁移学习还包括以下步骤：采用ResNet50作为预训练模型；利用反卷积网络的concate将预训练模型的输出做和，并利用1
×
1的卷积核将维度减少至N；使用softmax分类器进行分类，得到的变化图像的大小为w
×
h
×
N；其中，w表示图像的宽度、h表示图像的高度，N是定义的语义变化的类别数。6.根据权利要求1所述的一种基于小样本目标检测的轨道交通检测方法，其特征在于，所述S4中采用小样本目标检测算法对变化区域的具体形态进行识别还包括以下步骤：获取变化区域的图像和支持集，将变化区域的图像和支持集输入共享特征提取器，并生成查询和支持特征；将查询和支持特征输入密集关系蒸馏模块，并经过一个3
...

【专利技术属性】
技术研发人员：刘鹏，张真，汪良楠，张堃，曹骝，
申请(专利权)人：南京云创大数据科技股份有限公司，
类型：发明
国别省市：