[发明专利]一种基于对抗神经网络的电缆隧道异常检测方法

说明书

本发明提供一种基于对抗神经网络的电缆隧道异常检测方法，包括训练阶段和测试阶段，在所述训练阶段通过网络学习正常数据的数据分布，得到的模型A只能生成或者重建正常数据；在所述测试阶段，使用测试样本输入训练好的模型A，如果A经过重建后输出和输入接近，则为正常数据，否则为异常数据；本发明通过对抗神经网络技术，使检测到的隧道异常图片更真实，判别更准确，适于在隧道检测系统中推广。

技术领域

本发明涉及电缆隧道异常检测方法领域，尤其是基于对抗神经网络的电缆隧道异常检测方法。

背景技术

目前在做基于深度学习的图像异常分析时，首先收集足够多的异常样本数据，然后进行人工处理，将异常数据进行分类并对其中的异常区域进行标注，然后将这些人工处理好的图像数据输入给神经网络进行训练，最后再将需要识别的图像输入训练好的神经网络，即可得到图像是否为异常的结果；电缆隧道在大多数情况下都处于正常状态，即大多数情况下都只有正常样本，隧道中的异常情况很少出现，要积累异常样本数据就需要很长的时间，有些异常情况几年甚至10几年都不会发生一次，比如墙体开裂和火灾等，缺少异常样本数据，是目前深度学习所面临的一大挑战；目前巡检机器人采集可见光数据的主要方式还是录制视频，所以如果能够直接对视频进行分析处理是最理想的。

发明内容

为解决上述问题，本发明提供一种基于对抗神经网络的电缆隧道异常检测方法，可以将机器人拍摄的视频转换成图像，再通过对抗神经网络技术判别差异，提供更加准确的差异数据。

为解决上述技术问题，本发明的实施例提供一种基于对抗神经网络的电缆隧道异常检测方法，其特征在于，包括训练阶段和测试阶段，在所述训练阶段通过网络学习正常数据的数据分布，得到的模型A只能生成或者重建正常数据；在所述测试阶段，使用测试样本输入训练好的模型A，如果A经过重建后输出和输入接近，则为正常数据，否则为异常数据，其步骤如下：

（一）将标准视频转换为图像：从标准视频中提取图像序列，按固定时间段进行分类，每一图像类对应一个时间点，再从巡检视频中任取一幅图像，确定巡检图像与标准图像的对应关系，将视频转换为图像；

（二）将正常的标准图像转换成轮廓图：通过Canny 边缘检测算子对正常的标准图像进行边缘检测，得到轮廓图；

（三）将转换好的标准图像及其轮廓图输入到网络进行训练：对抗神经网络包括生成网络G（z）和判别网络D(x)，所述生成网络G（z）为生成图片的网络，其将轮廓图尽可能地生成为真实图像，所述判别网络D（x）判别图片真实性的网络，其输入x代表一张图片，输出D（x）代表x为真实图片的概率，所述概率为1，则为真实图片，否则为不真实图片；

其训练过程如下：

1)固定其中一个网络；

2)更新另一个网络的训练参数；

3)从所述网络数据中获得每次迭代需要使用的训练样本，对所述训练样本进行预处理后，作为模型A的输入；

4)采用随机梯度下降法交替优化判别网络和生成网络，更新各部分网络权值，直至迭代结束；

5)获得异常判别阈值；

（四）将日常巡检图像转换成轮廓图并输入到训练好的对抗神经网络中；将日常巡检图像通过Canny边缘检测算子进行边缘检测，得到轮廓图，输入到训练好的对抗神经网络中；

（五）找出两幅图像的差异部分：将生成的图像与真实的巡检图像做差值，如果差值大于异常判别阈值，则该区域为异常区域。

其中，步骤（三）中的所述生成网络G（z）输入给判别网络D（x）判别并得到相应的误差梯度，反向传播这些图片梯度成为组成生成网络G（z）的权重。

其中，步骤（三）中的所述训练参数包括最小批次值N batch 、最大迭代次数M、卷积和反卷积网络的层数以及每层网络的特征图数量。

其中，所述预处理包括图片尺寸归一化和像素值归一化。

其中，所述像素值归一化具体为：