[发明专利]一种基于NAS的岩石岩性识别方法、终端设备及存储介质

权利要求书

1.一种基于NAS的岩石岩性识别方法，其特征在于，包括以下步骤：

S1：采集岩石图片并对其岩性进行标注，根据标注后的岩石图片构建训练集；

S2：构建分类模型，分类模型采用基于NAS的网络模型作为主干网络，之后依次串联全卷积网络和分类网络；

S3：通过ImageNet图像数据集对分类模型进行预训练后，采用迁移学习算法，在分类模型中载入预训练后得到的权重参数后，再通过训练集对分类模型进行训练，训练后得到最终分类模型；

S4：通过最终分类模型对待识别的岩石图片中的岩性进行识别。

2.根据权利要求1所述的基于NAS的岩石岩性识别方法，其特征在于：步骤S1还包括对标注后的岩石图片进行数据增强处理，将数据增强处理后的所有岩石图片组成训练集。

3.根据权利要求2所述的基于NAS的岩石岩性识别方法，其特征在于：数据增强处理包括：尺寸放大或缩小、旋转、翻转、调整亮度、像素平移和添加噪声中的一者或多者。

4.根据权利要求1所述的基于NAS的岩石岩性识别方法，其特征在于：步骤S1还包括对组成训练集的岩石图片进行均衡化处理。

5.根据权利要求1所述的基于NAS的岩石岩性识别方法，其特征在于：分类模型的主干网络的网络结构依次包括：1个卷积层、7个MBConv层、1个卷积层、1个池化层和一个全连接层。

6.根据权利要求5所述的基于NAS的岩石岩性识别方法，其特征在于：MBConv层的网络结构依次包括：一个用于升维的1x1的卷积层、一个kxk的Depthwise卷积层、一个SE模块、一个用于降维的1x1的卷积层和一个Droupout层。

7.一种基于NAS的岩石岩性识别终端设备，其特征在于：包括处理器、存储器以及存储在所述存储器中并在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如权利要求1～6中任一所述方法的步骤。

8.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于：所述计算机程序被处理器执行时实现如权利要求1～6中任一所述方法的步骤。