一种基于NAS的岩石岩性识别方法、终端设备及存储介质与流程

一种基于nas的岩石岩性识别方法、终端设备及存储介质
技术领域
1.本发明涉及计算机视觉领域，尤其涉及一种基于nas的岩石岩性识别方法、终端设备及存储介质。


背景技术：

2.岩石岩性识别指通过某些方法认识和区别岩石的颜色、成分、结构等特征。岩性识别在油气勘探、矿产资源勘探等不同工作中都起着举足轻重的作用，比如岩性识别在油气勘探中保证着孔隙度和含油饱和度的准确性。因此在地质学研究中，如何对岩石岩性进行有效且高效的识别与分类一直是个亟待解决的重要课题。
3.传统的岩石分类方法主要包括以下两种：1.物理试验方法，即运用物理测试手段进行检测识别与分析。2.数学统计分析方法，即通过传统的数学统计与计算分析对岩石岩性分类特征进行识别与提取。这两类方法虽然效果良好，但是想要达到较高的准确率需要工作人员具有良好的实验理论和实验能力以及拥有完备的实验设备，而实际的地质勘探工作中一般都无法提供良好的实验环境，通常难以保证岩石岩性识别的实时性和准确性。


技术实现要素：

4.为了解决上述问题，本发明提出了一种基于nas的岩石岩性识别方法、终端设备及存储介质。
5.具体方案如下：
6.一种基于nas的岩石岩性识别方法，包括以下步骤：
7.s1：采集岩石图片并对其岩性进行标注，根据标注后的岩石图片构建训练集；
8.s2：构建分类模型，分类模型采用基于nas的网络模型作为主干网络，之后依次串联全卷积网络和分类网络；
9.s3：通过imagenet图像数据集对分类模型进行预训练后，采用迁移学习算法，在分类模型中载入预训练后得到的权重参数后，再通过训练集对分类模型进行训练，训练后得到最终分类模型；
10.s4：通过最终分类模型对待识别的岩石图片中的岩性进行识别。
11.进一步的，步骤s1还包括对标注后的岩石图片进行数据增强处理，将数据增强处理后的所有岩石图片组成训练集。
12.进一步的，数据增强处理包括：尺寸放大或缩小、旋转、翻转、调整亮度、像素平移和添加噪声中的一者或多者。
13.进一步的，步骤s1还包括对组成训练集的岩石图片进行均衡化处理。
14.进一步的，分类模型的主干网络的网络结构依次包括：1个卷积层、7个mbconv层、1个卷积层、1个池化层和一个全连接层。
15.进一步的，mbconv层的网络结构依次包括：一个用于升维的1x1的卷积层、一个kxk的depthwise卷积层、一个se模块、一个用于降维的1x1的卷积层和一个droupout层。
16.一种基于nas的岩石岩性识别终端设备，包括处理器、存储器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现本发明实施例上述的方法的步骤。
17.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现本发明实施例上述的方法的步骤。
18.本发明采用如上技术方案，与传统的机器视觉方法相比，具有以下优点：
19.1.模型通过搜索图像像素点提取物体特征，不需要手动提取待分类物体特征；
20.2.对于图像像素大小，成像距离及光照要求低；
21.3.采用适当的训练集可获得较好的识别分类效果，并具有良好鲁棒性和泛化能力。
附图说明
22.图1所示为本发明实施例一的流程图。
23.图2所示为该实施例中主干网络的网络结构示意图。
24.图3所示为该实施例中mbconv层的网络结构示意图。
具体实施方式
25.为进一步说明各实施例，本发明提供有附图。这些附图为本发明揭露内容的一部分，其主要用以说明实施例，并可配合说明书的相关描述来解释实施例的运作原理。配合参考这些内容，本领域普通技术人员应能理解其他可能的实施方式以及本发明的优点。
26.现结合附图和具体实施方式对本发明进一步说明。
27.实施例一：
28.本发明实施例提供了一种基于nas的岩石岩性识别方法，如图1所示，所述方法包括以下步骤：
29.s1：采集岩石图片并对其岩性进行标注，根据标注后的岩石图片构建训练集。
30.该实施例中采集的岩石图片包括七种岩性类别的图片，分别为：20张黑色煤图片、29张灰黑色泥岩图片、45张灰色泥质粉砂岩图片、17张灰色细砂岩图片、84张浅灰色细砂岩图片、39张深灰色粉砂质泥岩图片和74张深灰色泥岩图片。
31.由于采集的岩石图片数据有限，为了增加组成训练集的岩石图片的数量，提高模型训练的准确度，该实施例中还包括对对标注后的岩石图片进行数据增强处理，将数据增强处理后的所有岩石图片组成训练集。
32.该实施例中采用的数据增强处理方法包括：尺寸放大或缩小、旋转(任意角度，如45
°
，90
°
，180
°
，270
°
)、翻转(水平翻转，垂直翻转)、明亮度改变(变亮，变暗)、像素平移(往一个方向平移像素，空出部分自动填补黑色)、添加噪声(椒盐噪声，高斯噪声)等，在其他实施例中也可以采用其他数据增强方式，在此不做限定。
33.进一步的，该实施例中还包括对数据增强后的岩石图片进行均衡化处理，以增强图片对比度，进而提高模型训练结果的准确度。
34.s2：构建分类模型，分类模型采用基于nas(神经网络结构搜索)的网络模型作为主干网络，之后依次串联全卷积网络(fcn)和基于softmax函数的分类网络。
35.主干网络的网络结构如图2所示，从输入端开始，首先设置1个卷积层，7个mbconv层；再设置1个1x1的卷积层，连接1个池化层，最后连接fc层；其中mbconv层的网络结构如图3所示，主要由一个1x1的普通卷积(升维作用，包含bn和swish)，一个kxk的depthwiseconv卷积(包含bn和swish)，一个se模块，一个1x1的普通卷积(降维作用，包含bn)，一个droupout层构成。
36.s3：通过imagenet图像数据集对分类模型进行预训练后，采用迁移学习算法，在分类模型中载入预训练后得到的权重参数后，再通过训练集对分类模型进行训练，训练后得到最终分类模型。
37.imagenet图像数据集为公开数据集。
38.预训练得到分类模型的一组权重参数，将该组权重参数迁移至分类模型中，可以加速整体训练进程。分类模型载入预训练的权重参数后，通过步骤s1构建的训练集对分类模型中的所有参数进行训练，通过反复迭代训练得到最终分类模型。该实施例中设定迭代次数为100，学习率为0.01。
39.s4：通过最终分类模型对待识别的岩石图片中的岩性进行识别。
40.通过模拟实验，本实施例分类模型的准确率能够达到90％，因此本实施例分类模型训练效果良好，能对七类岩石实现较为精准的识别。
41.选取七类岩石图片中的各一张进行岩石岩性识别，结果如表1所示。
42.表1
[0043][0044]
与岩石岩性的识别与分类的两类传统方法相比，本发明实施例采用基于nas的分类模型，通过深度学习算法分析处理图像中的岩石特征并建立相应数学模型，从而利用图像识别技术实现岩石岩性识别与分类，一定程度上保证了岩石岩性识别的实时性和准确性，在实际工作中更具实用性。
[0045]
实施例二：
[0046]
本发明还提供一种基于nas的岩石岩性识别终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现本发明实施例一的上述方法实施例中的步骤。
[0047]
进一步地，作为一个可执行方案，所述基于nas的岩石岩性识别终端设备可以是桌上型计算机、笔记本、掌上电脑及云端服务器等计算设备。所述基于nas的岩石岩性识别终端设备可包括，但不仅限于，处理器、存储器。本领域技术人员可以理解，上述基于nas的岩石岩性识别终端设备的组成结构仅仅是基于nas的岩石岩性识别终端设备的示例，并不构
成对基于nas的岩石岩性识别终端设备的限定，可以包括比上述更多或更少的部件，或者组合某些部件，或者不同的部件，例如所述基于nas的岩石岩性识别终端设备还可以包括输入输出设备、网络接入设备、总线等，本发明实施例对此不做限定。
[0048]
进一步地，作为一个可执行方案，所称处理器可以是中央处理单元(central processing unit，cpu)，还可以是其他通用处理器、数字信号处理器(digital signal processor，dsp)、专用集成电路(application specific integrated circuit，asic)、现场可编程门阵列(field-programmable gate array，fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等，所述处理器是所述基于nas的岩石岩性识别终端设备的控制中心，利用各种接口和线路连接整个基于nas的岩石岩性识别终端设备的各个部分。
[0049]
所述存储器可用于存储所述计算机程序和/或模块，所述处理器通过运行或执行存储在所述存储器内的计算机程序和/或模块，以及调用存储在存储器内的数据，实现所述基于nas的岩石岩性识别终端设备的各种功能。所述存储器可主要包括存储程序区和存储数据区，其中，存储程序区可存储操作系统、至少一个功能所需的应用程序；存储数据区可存储根据手机的使用所创建的数据等。此外，存储器可以包括高速随机存取存储器，还可以包括非易失性存储器，例如硬盘、内存、插接式硬盘，智能存储卡(smart media card,smc)，安全数字(secure digital,sd)卡，闪存卡(flash card)、至少一个磁盘存储器件、闪存器件、或其他易失性固态存储器件。
[0050]
本发明还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现本发明实施例上述方法的步骤。
[0051]
所述基于nas的岩石岩性识别终端设备集成的模块/单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本发明实现上述实施例方法中的全部或部分流程，也可以通过计算机程序来指令相关的硬件来完成，所述的计算机程序可存储于一计算机可读存储介质中，该计算机程序在被处理器执行时，可实现上述各个方法实施例的步骤。其中，所述计算机程序包括计算机程序代码，所述计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。所述计算机可读介质可以包括：能够携带所述计算机程序代码的任何实体或装置、记录介质、u盘、移动硬盘、磁碟、光盘、计算机存储器、只读存储器(rom，read-only memory)、随机存取存储器(ram，random access memory)以及软件分发介质等。
[0052]
尽管结合优选实施方案具体展示和介绍了本发明，但所属领域的技术人员应该明白，在不脱离所附权利要求书所限定的本发明的精神和范围内，在形式上和细节上可以对本发明做出各种变化，均为本发明的保护范围。