一种基于CNN+RNN算法的三维探地雷达图像地下管线识别方法与流程

一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法
技术领域
1.本发明属于探地雷达三维回波图后处理的目标检测领域，具体地，涉及一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法。


背景技术：

2.探地雷达是一种高效率的、无破坏性的用于探测浅层地下环境的探测技术。探地雷达通过发射天线不断向地层发射高频电磁波，电磁波可以穿透地下介质，但是由于不同的地下介质具有不同的介电常数，电磁波在地层传播时，会在介质交替层发生反射和折射。接收天线通过接收多道反射回波(a-scan)信号，通过信号处理技术，拼接为反映地下结构的二维b-scan图像，但是二维b-scan图像并不能完全反映地下目标的特征。容易受到检测方向、图像选取方式等情况的影响，单独从二维b-scan图像中的获取信息非常容易而发生漏检、误检的情况。如果采用的阵列式发射天线和接收天线，每个天线将会得到一个平面的二维b-scan图像，通过将空间内的多个二维b-scan图像按顺序堆叠，将会得到一副可以完整反映地下空间结构的三维图像(c-scan)。
3.传统的检测方法是通过人工分辨的方式，专业技术人员根据能量、同相性以及波形等先验经验来分辨地层结构和被掩埋物体。由于地下环境的复杂性，例如地下物体一般具有复杂的几何形状，再加上地下介质水分含量的变化和地下其他物体的干扰等影响，使得探地雷达的数据解释具有极强的专业性，同时人工识别的效率较低，一名专家通常需要几天的时间才能分析几公里的探地雷达数据，这种方法效率低下且常常导致漏检或误检的问题。现如今深度学习发展火热，各式各样的神经网络模型相继出现，例如cnn通过卷积神经网络逐层提取二维图像上的特征来实现目标检测，但是这些网络模型基本都是根据特定的二维输入图像来设计的，目的是逐层提取二维图像上的特征来实现目标检测。由于探地雷达二维回波图像并不能完全反映地下目标特征，在某些情况下可能会发生漏检和误检测情况，如何利用地下结构的三维图像来准确识别目标具有重要意义。


技术实现要素：

4.本发明提出了一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法，用以解决现有方法难以检测识别三维探地雷达图像中地下管线的问题，减少漏检和误检测的情况。
5.本发明是通过以下方案实现的：
6.一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法：
7.所述方法具体包括以下步骤：
8.步骤1：通过探地雷达获得三维回波图像，对三维回波图像进行预处理，所述预处理包括直达波去除和小波变换；
9.步骤2：将已预处理的探地雷达的三维回波图像进行标注，分别为横向管线、纵向
管线、地下空洞和无目标四类，并将其打乱，随机分配至训练集和验证集；
10.步骤3：利用步骤2的训练集和验证集对cnn rnn的神经网络模型进行训练，得到训练好的权重模型；
11.所述cnn rnn的神经网络模型结构为：先将三维图像中的各个二维图像分别送入cnn中，利用cnn来对三维图像的各个二维图像进行特征提取，然后再将同一幅三维图像的各个二维图像的特征进行整合，一起送到rnn网络中对这些特征进行处理，最后分类识别出目标物体；
12.cnn模型结构为：经过第一层的8个(3，3)的卷积核进行二维卷积操作，然后采用批标准化对每一批的训练数据进行归一化，再采用relu激活函数完成数据的非线性变换；第二层为16个(3，3)卷积核的二维卷积操作，采用批标准化和relu激活函数，并进行(2，2)的最大池化操作；第三层为32个(3，3)卷积核的二维卷积操作，其余与第一层的参数和步骤相同；第四层为64个(3，3)卷积核的二维卷积操作，与第二层一样采用批标准化、relu激活函数和池化，并进行0.2的dropout，随机删除训练过程中的一部分的隐藏神经元，用来减少训练时间和减少过拟合；
13.在cnn网络后添加一个全连接层，将提取好的特征进行分类为36类，再由flatten层展平提取到的特征向量送入rnn中；
14.rnn采用单层的lstm递归神经网络；rnn部分的lstm网络隐藏单元的维度设置为36；最后送入全连接层，对应连接到四个节点，经过softmax分类输出；
15.lstm递归神经网络模型含有三个门限：输入门i
t
、遗忘门f
t
、输出门o
t
，拥有表征短期记忆的记忆体h
t
，表征长时记忆的细胞态c
t
以及长时记忆的候选态
16.i
t
＝σ(wi·
[h
t-1
,x
t
] bi)
[0017]ft
＝σ(wf·
[h
t-1
,x
t
] bf)
[0018]ot
＝σ(wo·
[h
t-1
,x
t
] bo)
[0019][0020]ht
＝o
t
*tanh(c
t
)
[0021][0022]
其中wi、wf、wo、wc分别是对应门的权重矩阵，bi、bf、bo、bc是相应的偏置，σ表示sigmoid函数。
[0023]
步骤4：利用步骤3获得训练好的神经网络模型，对探地雷达三维回波图像进行管线目标识别检测，最终标注出带有地下管线信息的三维回波图像。
[0024]
进一步地，在步骤1中，
[0025]
利用时间门限截取的方式进行直达波去除来提高信噪比，将回波数据b-scan图像中包含直达波横线状的部分删除，不减弱横线管道的特征，保留地下管线的双曲线特征。
[0026]
进一步地，在步骤1中，
[0027]
将探地雷达的三维回波图像数据y(k)简化表示为y(k)＝x(k) n(k)，
[0028]
其中x(k)为地下目标的有用数据，n(k)表示噪声分量；
[0029]
对y(k)进行小波变换来提取有用的信号分量，公式为y
i，j
＝x
i，j
n
i，j
，其中x
i，j
即目标回波信号的小波变换的小波系数，n
i，j
为噪声分量小波变换后的小波系数；
[0030]
经小波变换后，目标回波信号的小波系数x
i，j
幅值大于噪声分析的小波系数n
i，j
；设置一个临界阈值，当小波系数小于该阈值时，认为它是噪声分量，并将其设置为0；当小波系数大于该阈值时，则可以认为它是有用信号的小波系数，保留这部分系数。处理后再进行小波逆变换，经过小波重构得到去噪后的探地雷达三维回波图像。
[0031]
进一步地，在步骤2中，
[0032]
对步骤1中得到的三维回波图像进行标签标注，按照4：1的比例随机分派给训练集和验证集，数据集共有4个分类，分别为横向管线、纵向管线、地下空洞和无目标；
[0033]
其中当管线和探地雷达测道方向的夹角小于45
°
时，标记为纵向管线，当管线和探地雷达测道方向的夹角大于45
°
时，标记为横向管线。
[0034]
进一步地，在步骤3中，
[0035]
通过所述步骤2中生成的训练集对cnn rnn的神经网络模型进行训练，训练参数为10batch和10epochs，学习率为0.0001。最终得到经过训练的权重模型。
[0036]
进一步地，在步骤4中，
[0037]
利用训练好的神经网络模型，对探地雷达的三维回波图像进行地下管线目标检测；将未输入过该系统的地下管线目标探地雷达三维回波图像输入到步骤3训练好的神经网络模型中，自动地对探地雷达三维回波图像中的地下管线目标进行检测，最终标注出带有地下管线信息的三维回波图像。
[0038]
本发明有益效果
[0039]
(1)本发明通过已经人工分类出的地下管线标签的三维图像作为训练集，利用该训练集训练出可用于自动识别后续探地雷达三维回波图像中的地下管线信息的神经网络模型。采用本发明的方法对探地雷达三维回波图像的地下管线目标进行检测可以有效提高识别概率和识别速度；
[0040]
(2)实际中的探地雷达采集到的数据，由于地下环境的复杂性，仅依靠二维b-scan图像，无法表现出地下目标的全貌，这使后续目标的识别很容易出现误判和漏判的情况。本发明的目的是将探地雷达的三维回波图像分为多个b-scan图像，再将其输入到卷积神经网络和循环神经网络中去训练网络模型，利用卷积神经网络对三维回波图像的各个b-scan图像进行特征提取，再利用对这些特征进行整合，再送入到rnn网络中对这些特征处理，最后分类识别出目标物体。最终将训练好的模型用于实现探地雷达三维回波图像地下管线目标的识别；本发明可以将地下管线目标识别概率提高到95％以上。
附图说明
[0041]
图1为本发明的基于cnn rnn算法的三维探地雷达图像地下管线识别方法的流程图；
[0042]
图2为采用的cnn rnn模型结构图；
[0043]
图3为本发明的直达波去除前后的对比图；
[0044]
图4为本发明的横向管线和纵向管线的b-scan图像；
[0045]
图5为本发明的地下管线和地下管线的b-scan图像和反映它们的空间结构图像；
[0046]
图6为本发明lstm递归神经网络模型；
[0047]
图7为本发明网络模型训练过程中的准确率曲线图和损失函数曲线图。
具体实施方式
[0048]
下面将结合本发明实施例中的附图对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例；基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。
[0049]
结合图1至图7，
[0050]
一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法：
[0051]
所述方法具体包括以下步骤：
[0052]
步骤1：通过探地雷达获得三维回波图像，对三维回波图像进行预处理，所述预处理包括直达波去除和小波变换；
[0053]
步骤2：将已预处理的探地雷达的三维回波图像进行标注，分别为横向管线、纵向管线、地下空洞和无目标四类，并将其打乱，随机分配至训练集和验证集；
[0054]
步骤3：利用步骤2的训练集和验证集对cnn rnn的神经网络模型进行训练，得到训练好的权重模型；
[0055]
步骤4：利用步骤3获得训练好的神经网络模型，对探地雷达三维回波图像进行管线目标识别检测，最终标注出带有地下管线信息的三维回波图像。
[0056]
在步骤1中，
[0057]
所述步骤1利用时间门限截取的方式进行直达波滤除，是将回波数据b-scan图像中包含直达波横线状的部分直接截掉。想要对目标进行精确的识别，首要目标是提高信噪比。而直达波属于探地雷达回波图像中最主要的干扰来源之一。所以去除该部分回波信号有助于提高对地下管线识别率。采用时间门限截取的方式是最直接、最有效对直达波去除的方式，而且针对土地介质交界处不是水平的情况仍有很好的滤除效果，也不会减弱横线管道的特征，能很好地保留地下管线的双曲线特征。
[0058]
对于直达波的抑制是探地雷达信号处理中一个重要环节。如图3所示，由于直达波相对有用信号来说能量很强，有用信号很容易被覆盖而无法分辨。想要对目标进行精确的识别，首要目标是提高信噪比。而直达波属于探地雷达回波图像中最主要的干扰来源之一，直达波会对神经网络提取目标特征造成干扰。所以去除该部分回波信号有助于提高对地下管线识别率。
[0059]
在步骤1中，
[0060]
将探地雷达的三维回波图像数据y(k)简化表示为y(k)＝x(k) n(k)，
[0061]
其中x(k)为地下目标的有用数据，n(k)表示噪声分量；
[0062]
为了尽可能的从y(k)中提取有用的信号分量，对探地雷达的回波数据进行小波变换，公式为y
i，j
＝x
i，j
n
i，j
，其中x
i，j
即目标回波信号的小波变换的小波系数，n
i，j
为噪声分量小波变换后的小波系数；
[0063]
经小波变换后，目标回波信号的小波系数x
i，j
幅值大于噪声分析的小波系数n
i，j
；设置一个临界阈值，当小波系数小于该阈值时，认为它是噪声分量，并将其设置为0；当小波系数大于该阈值时，则可以认为它是有用信号的小波系数，保留这部分系数。处理后再进行小波逆变换，经过小波重构得到去噪后的探地雷达三维回波图像。
[0064]
未经处理时的信号有时极小，有时还被直达波覆盖而不能分辨，而且还会受到噪
声等影响，使得数据图像不易分辨，这样在运用神经网络进行特征提取分类时很难保证准确率。
[0065]
在步骤2中，
[0066]
对步骤1中得到的三维回波图像进行标签标注，按照4：1的比例随机分派给训练集和验证集，数据集共有4个分类，分别为横向管线、纵向管线、地下空洞和无目标；
[0067]
其中当管线和探地雷达测道方向的夹角小于45
°
时，标记为纵向管线，当管线和探地雷达测道方向的夹角大于45
°
时，标记为横向管线。
[0068]
当探地雷达的测线方向与管线方向垂直时，可以看到b-scan图像呈现双曲线形状；当探地雷达的测线方向与管线方向相同时，b-scan图像上并不会呈现双曲线形状，如图4所示。这是由于探地雷达记录天线正下方测量反射信号的强度及反射设计，因此，当探地雷达先靠近然后远离埋藏的物体时，反射在雷达扫描中形成双曲线。双曲线的顶点可以计算出目标物体的实际位置。而当测道方向与管线方向相同时，就没有这样一个先靠近再远离的过程，显然这时就不会再b-scan中出现双曲线的形状特征。并且对于地下管线和空洞，它们在b-scan图像上都会呈现出双曲线状的回波图像，难以辨认区分，如图5所示。这样就对传统的基于二维图像的卷积神经网络提取特征造成了一定的困难，非常容易发生漏判和误判。
[0069]
在步骤3中，
[0070]
所述cnn rnn的神经网络模型结构为：先将三维图像中的各个二维图像分别送入cnn中，利用cnn来对三维图像的各个二维图像进行特征提取，然后再将同一幅三维图像的各个二维图像的特征进行整合，一起送到rnn网络中对这些特征进行处理，最后分类识别出目标物体；
[0071]
cnn模型结构为：经过第一层的8个(3，3)的卷积核进行二维卷积操作，然后采用批标准化对每一批的训练数据进行归一化，再采用relu激活函数完成数据的非线性变换；第二层为16个(3，3)卷积核的二维卷积操作，采用批标准化和relu激活函数，并进行(2，2)的最大池化操作；第三层为32个(3，3)卷积核的二维卷积操作，其余与第一层的参数和步骤相同；第四层为64个(3，3)卷积核的二维卷积操作，与第二层一样采用批标准化、relu激活函数和池化，并进行0.2的dropout，随机删除训练过程中的一部分的隐藏神经元，用来减少训练时间和减少过拟合；
[0072]
由于一层lstm参数量是同样输入和输出数目的cnn参数量的四倍多，如果直接将cnn网络提取到的特征图直接输入lstm网络中会产生大量参数，所以在cnn网络后添加一个全连接层，将提取好的特征进行分类为36类，再由flatten层展平提取到的特征向量送入rnn中；
[0073]
rnn采用单层的lstm递归神经网络；rnn部分的lstm网络隐藏单元的维度设置为36；最后送入全连接层，对应连接到四个节点，经过softmax分类输出；
[0074]
lstm(长短期记忆)递归神经网络模型如图6所示，含有三个门限：输入门i
t
、遗忘门f
t
、输出门o
t
，拥有表征短期记忆的记忆体h
t
，表征长时记忆的细胞态c
t
以及长时记忆的候选态
[0075]it
＝σ(wi·
[h
t-1
,x
t
] bi)
[0076]ft
＝σ(wf·
[h
t-1
,x
t
] bf)
[0077]ot
＝σ(wo·
[h
t-1
,x
t
] bo)
[0078][0079]ht
＝o
t
*tanh(c
t
)
[0080][0081]
其中wi、wf、wo、wc分别是对应门的权重矩阵，bi、bf、bo、bc是相应的偏置，σ表示sigmoid函数。一个lstm单元会引入四组不同的权重和偏置，参数会比cnn中的神经元模型更多。
[0082]
在步骤3中，
[0083]
通过所述步骤2中生成的训练集对cnn rnn的神经网络模型进行训练，训练参数为10batch和10epochs，学习率为0.0001。迭代得到的识别准确率曲线和损失曲线如图7所示，可以看到，训练迭代到6个周期后准确率和损失函数逐渐稳定下来，训练和验证的结果基本一致，没有出现明显的过拟合。最终得到经过训练的权重模型。
[0084]
在步骤4中，
[0085]
利用训练好的神经网络模型，对探地雷达的三维回波图像进行地下管线目标检测；将未输入过该系统的地下管线目标探地雷达三维回波图像输入到步骤3训练好的神经网络模型中，自动地对探地雷达三维回波图像中的地下管线目标进行检测，最终标注出带有地下管线信息的三维回波图像。
[0086]
以上对本发明所提出的一种基于cnn rnn算法的三维探地雷达图像地下管线识别方法，进行了详细介绍，对本发明的原理及实施方式进行了阐述，以上实施例的说明只是用于帮助理解本发明的方法及其核心思想；同时，对于本领域的一般技术人员，依据本发明的思想，在具体实施方式及应用范围上均会有改变之处，综上所述，本说明书内容不应理解为对本发明的限制。