一种边坡岩石类别自动识别的方法技术

本发明专利技术的一种边坡岩石类别自动识别的方法，岩质边坡岩石的分类对于边坡稳定性的分析至关重要，目前人工方法效率低且受主观因素影响，建立岩质边坡图像集分析的卷积神经网络模型，通过卷积操作和池化操作分别对80000张岩质边坡图像进行特征信息的提取和压缩，然后对网络模型进行训练从而实现了岩质边坡岩石的自动识别与分类；采用训练集和测试集中的岩质边坡图像对模型进行检验分析，训练集准确率达到了98％，测试集准确率达到了90％，显示了训练之后的网络模型具有良好的鲁棒性，能达到理想的训练效果。最终采用深度学习建立的网络模型，实现了岩质边坡岩石识别的快速化、自动化。

A Method for Automatic Recognition of Slope Rock Classes

【技术实现步骤摘要】
一种边坡岩石类别自动识别的方法
本专利技术涉及建设工程信息化
，具体是一种边坡岩石类别自动识别的方法。
技术介绍
边坡不同岩石的类别和范围的划定是边坡科学研究中的基础性工作。过去往往通过复杂的仪器设备进行现场取样，人为的根据岩石的颜色、结构构造等辨识边坡岩石的种类以及不同岩石边界范围的标定，这样的过程费时费力。常规的边坡岩石识别与分类主要有两种方法。第一种是物理试验法，即运用物理测试手段对边坡岩石进行检测。例如，采用X射线粉末衍射、扫描电镜、红外光谱、差热分析、电子探针、高光谱图像等方法进行分析。第二种是数学统计分析法，即通过传统的数学统计与计算分析对岩质边坡的分类特征进行识别与提取。例如，运用Sr和Yb作为分类特征或者综合运用多重分形局部奇异性与空间加权分析的方法识别和提取岩石异常信息。上述两类方法存在实验复杂、周期偏长、受主观影响较大等缺陷，同时无法满足现场测量评价的需要，为了可以快速准确的提取岩质边坡岩石信息，很多专家已经通过智能算法对岩石图像进行了研究：张旭等应用朴素贝叶斯K邻近算法进行了岩石图像分类；康丽萍等分别利用softmax多分类器和多分类支持向量机(SVM)实现图像分类；张嘉凡等提出了基于聚类分析算法的岩石CT图像分割及量化方法；Lietal.采用迁移学习方法对砂岩图像进行了训练，最终获得了精度较高的砂岩图像分类模型。分析上述研究进展发现存在一些不足之处：图像采用经过后续加工后的标准岩石薄片，而不是以原始状态的岩质边坡图像作为训练数据集；其次，未对边坡不同岩石的边界进行标定，无法确定边坡各类岩石的边界范围。
技术实现思路
为解决上述技术问题，本专利技术的目的是提供一种边坡岩石类别自动识别的方法，实现了岩质边坡岩石识别与边界范围划分的快速化、自动化。本专利技术提供一种边坡岩石类别自动识别的方法，包括：步骤1：采集边坡岩石图像，并对图像进行处理，建立边坡岩石图像样本库；步骤2：建立卷积神经网络，隐藏层包括13个卷积层和5个池化层；步骤3：对神经网络进行训练，将处理后的图像输入到卷积神经网络的卷积层进行特征提取，通过得分函数计算每张图像属于各类别的得分值，进行分类预测；步骤4：卷积之后的特征图输入到池化层进行压缩；步骤5：经过神经网络的反复卷积核池化操作之后，通过SVM损失函数对分类预测进行修正，通过softmax分类器将分类得分值转换成为分类概率值，进而输出图像的预测类别；步骤6：将预测类别和实际类别做差，使用极小化误差的方法反向传播优化各权值参数，经过多次迭代后对训练进行评价，直到达到最大迭代次数，完成训练；步骤7：通过训练好的神经网络对边坡岩石图像进行分类。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤1包括：步骤1.1：采用无人机设备对岩质边坡全景图像进行采集，获取高分辨率原始图像集；步骤1.2：对图像进行增强处理；步骤1.3：采用Horizontalflips操作，对图像进行水平翻转；步骤1.4：再采用Randomcrops操作，对图像进行裁剪，将图片调整为224×224的大小，构成图像样本库。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤2中构建的卷积神经网络包括依次连接的输入层、第一卷积层、第二卷积层、第一池化层、第三卷积层、第四卷积层、第二池化层、第五卷积层、第六卷积层、第七卷积层、第三池化层、第八卷积层、第九卷积层、第十卷积层、第四池化层、第十一卷积层、第十二卷积层、第十三卷积层、第五池化层和全连接层。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤3中得分函数为：式中，f(x,y)表示边坡岩石图像上点(x,y)的灰度值，w(x,y)表示卷积核，在卷积之前图像输入的大小为W1×H1×D1，卷积过程中需要指定的超参数为卷积核filter的个数K，filter的大小F，stride步长S，pad边界填充P，经过一次卷积操作后图像输出的大小为W2×H2×D2：W2＝(W1-F+2P)/S+1H2＝(H1-F+2P)/S+1D2＝K使用ReLU函数作为激活函数：f(x)＝max(0,x)，当x≤0时，当x＞0时，在本专利技术的边坡岩石类别自动识别的方法中，所述步骤4中采用最大池化计算方法，池化核大小设定为2×2，stride取2。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤5中SVM损失函数为：其中，sj表示预测结果属于错误类别的得分，表示预测结果属于正确类别时的得分。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤5中softmax函数为：其中，输入值是一组向量，向量中各元素为所有类别的评分值，结果输出一组向量，向量中每个元素值在0到1之间，且所有元素之和为1，式中Zj表示边坡岩石图像属于正确类别时的得分，Zk表示图像属于错误类别时的得分。在本专利技术的边坡岩石类别自动识别的方法中，所述步骤5中得分值转化为概率值时采用如下式所示的损失函数：式中，fyi表示边坡岩石图像属于正确类别时的概率值，fj表示图像属于错误类别时的概率值。本专利技术的一种边坡岩石类别自动识别的方法，至少具有以下有益效果：利用无人机设备对边坡进行全景高清图像采集，极大地满足卷积神经网络的训练要求。将卷积神经网络分类操作应用到边坡岩石图像分类中，输入原始岩质边坡图像数据，由卷积网络自动提取图像特征，训练之后网络达到90％的准确率，网络模型可以有效的提取边坡岩石图像的特征，实现岩质边坡岩石的自动识别分类。以岩石的颜色作为主要区分特征，利用深度学习回归操作对边坡不同类别岩石的范围进行划分。输入任意边坡图像进行检测，验证了网络模型边界范围确定方法的准确性，实现了岩质边坡不同岩石划分快速化、自动化的要求。附图说明图1是本专利技术的一种边坡岩石类别自动识别的方法流程图；图2是本专利技术采用的卷积神经网络的模型结构图；图3是最大池化计算方式图；图4是Drop-Out计算方法。具体实施方式如图1所示，本专利技术的一种边坡岩石类别自动识别的方法，包括：步骤1：采集边坡岩石图像，并对图像进行处理，建立边坡岩石图像样本库，所述步骤1包括：步骤1.1：采用无人机设备对岩质边坡全景图像进行采集，获取高分辨率原始图像集；步骤1.2：对图像进行增强处理；步骤1.3：采用Horizontalflips操作，对图像进行水平翻转；步骤1.4：再采用Randomcrops操作，对图像进行裁剪，将图片调整为224×224的大小，构成图像样本库。对高陡边坡岩石图像的采集存在一定的风险性，复杂多变的地质条件加大了工程人员的测量难度，如何不受地质条件的限制实现边坡全景图像的获取，并保证测量人员的安全，成为边坡图像获取中的棘手问题。对此采用无人机设备进行了岩质边坡全景图像的采集，取得了很好的效果。生成8000张高分辨率原始图像数据集，对数据集进行增强处理，先将8000张原始数据集图片采用Horizontalflips操作，对图像进行水平翻转，再采用Randomcrops操作，对图像进行裁剪，将图片调整为224×224的大小，图像数据集扩大到80000张；采集了常见的四类边坡岩石图像，训练集与测试集数量见表1，训练集是从各类岩质边坡总样本中随机抽取，剩下的作为测试集。表1样本数据步骤2：卷积神经网络为代表的深度学习模型是目前图像识别中应用最广泛的方法，完整的卷积神经网络本文档来自技高网...

【技术保护点】
1.一种边坡岩石类别自动识别的方法，其特征在于，包括：步骤1：采集边坡岩石图像，并对图像进行处理，建立边坡岩石图像样本库；步骤2：建立卷积神经网络，隐藏层包括13个卷积层和5个池化层；步骤3：对神经网络进行训练，将处理后的图像输入到卷积神经网络的卷积层进行特征提取，通过得分函数计算每张图像属于各类别的得分值，进行分类预测；步骤4：卷积之后的特征图输入到池化层进行压缩；步骤5：经过神经网络的反复卷积核池化操作之后，通过SVM损失函数对分类预测进行修正，通过softmax分类器将分类得分值转换成为分类概率值，进而输出图像的预测类别；步骤6：将预测类别和实际类别做差，使用极小化误差的方法反向传播优化各权值参数，经过多次迭代后对训练进行评价，直到达到最大迭代次数，完成训练；步骤7：通过训练好的神经网络对边坡岩石图像进行分类。

【技术特征摘要】
1.一种边坡岩石类别自动识别的方法，其特征在于，包括：步骤1：采集边坡岩石图像，并对图像进行处理，建立边坡岩石图像样本库；步骤2：建立卷积神经网络，隐藏层包括13个卷积层和5个池化层；步骤3：对神经网络进行训练，将处理后的图像输入到卷积神经网络的卷积层进行特征提取，通过得分函数计算每张图像属于各类别的得分值，进行分类预测；步骤4：卷积之后的特征图输入到池化层进行压缩；步骤5：经过神经网络的反复卷积核池化操作之后，通过SVM损失函数对分类预测进行修正，通过softmax分类器将分类得分值转换成为分类概率值，进而输出图像的预测类别；步骤6：将预测类别和实际类别做差，使用极小化误差的方法反向传播优化各权值参数，经过多次迭代后对训练进行评价，直到达到最大迭代次数，完成训练；步骤7：通过训练好的神经网络对边坡岩石图像进行分类。2.如权利要求1所述的边坡岩石类别自动识别的方法，其特征在于，所述步骤1包括：步骤1.1：采用无人机设备对岩质边坡全景图像进行采集，获取高分辨率原始图像集；步骤1.2：对图像进行增强处理；步骤1.3：采用Horizontalflips操作，对图像进行水平翻转；步骤1.4：再采用Randomcrops操作，对图像进行裁剪，将图片调整为224×224的大小，构成图像样本库。3.如权利要求1所述的边坡岩石类别自动识别的方法，其特征在于，所述步骤2中构建的卷积神经网络包括依次连接的输入层、第一卷积层、第二卷积层、第一池化层、第三卷积层、第四卷积层、第二池化层、第五卷积层、第六卷积层、第七卷积层、第三池化层、第八卷积层、第九卷积层、第十卷积层、第四池化层、第十一卷积层、第...

【专利技术属性】
技术研发人员：王述红，王鹏宇，朱承金，张紫杉，
申请(专利权)人：东北大学，
类型：发明
国别省市：辽宁,21