[发明专利]一种岩石图像分类方法、系统、设备及介质

权利要求书

1.一种岩石图像分类方法，其特征在于，包括：

构建岩石图像分类模型；

训练所述岩石图像分类模型；

采集目标岩石图像，输入所述目标岩石图像至训练好的岩石图像分类模型，预测岩石类别。

2.根据权利要求1所述的岩石图像分类方法，其特征在于，所述构建岩石图像分类模型的步骤包括：

选取残差网络Resnet50作为主干网络；

在残差网络Resnet50中融合注意力机制，得到岩石图像分类模型。

3.根据权利要求2所述的岩石图像分类方法，其特征在于，所述输入所述目标岩石图像至训练好的岩石图像分类模型，预测岩石类别的步骤包括：

利用残差网络提取目标岩石图像中多个特征图；

采用注意力机制学习每个特征图的权重；

将目标岩石图像中每个特征图的权重与特征图的原始特征相乘，得到多种岩石特征。

4.根据权利要求1所述的岩石图像分类方法，其特征在于，所述构建岩石图像分类模型的步骤包括：

选取残差网络ResNet-18作为主干网络；

在残差网络ResNet-18中融合卷积神经网络，得到岩石图像分类模型。

5.根据权利要求4所述的岩石图像分类方法，其特征在于，所述输入所述目标岩石图像至训练好的岩石图像分类模型，预测岩石类别的步骤包括：

利用残差网络提取目标岩石图像中多个特征图；

采用卷积神经网络方法提取每个特征图中岩石特征，得到多种类型岩石。

6.根据权利要求3或5所述的岩石图像分类方法，其特征在于，所述利用残差网络提取目标岩石图像中多个特征图的步骤包括：

将所述目标岩石图像输入所述残差网络进行卷积处理，得到多个特征图；

对多个特征图进行批量归一化处理，并线性修正多个特征图。

7.根据权利要求1所述的岩石图像分类方法，其特征在于，所述训练岩石图像分类模型的步骤包括：

建立岩石图像样本库，其中所述岩石图像样本库包括多种岩石样本图像数据；

拆分所述岩石图像样本库，形成训练集和测试集，其中所述训练集为第一部分岩石样本图像数据；所述测试集为第二部分岩石样本图像数据；

根据所述训练集训练所述岩石图像分类模型，得到预训练的岩石图像分类模型；

输入所述测试集至预训练的岩石图像分类模型，确定评价参数；

确定所述评价参数符合预设要求；

基于所述评价参数符合预设要求，确定训练好的岩石图像分类模型。

8.一种岩石图像分类系统，其特征在于，包括：

模型构建模块，用于构建岩石图像分类模型；

模型训练模块，用于训练岩石图像分类模型；

分类模块，用于采集目标岩石图像，输入所述目标岩石图像至训练好的岩石图像分类模型，预测岩石类别。

9.一种岩石图像分类设备，其特征在于，包括：

存储器，用于存储计算机程序；

处理器，用于执行所述计算机程序时实现如权利要求1至7任一项所述岩石图像分类方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至7任一项所述岩石图像分类方法的步骤。