[发明专利]一种岩石识别方法、装置、设备及存储介质

权利要求书

1.一种岩石识别方法，其特征在于，所述方法包括：

对待识别岩石的图片进行特征抽取，得到特征向量；

利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，得到中间分类特征向量；所述预设岩石分类空间用于对所述特征向量进行中心点约束；其中，不同中心点对应岩石的不同细分类；

根据所述中间分类特征向量，计算所述待识别岩石属于不同岩石类别的概率，得到概率分布向量；

根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别；

其中：

利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，包括：

将所述特征向量映射到所述预设岩石分类空间中，得到所述特征向量在所述预设岩石分类空间中的第一投射点；

计算所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；

根据所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离，确定与所述特征向量距离最近的目标中心点；

根据所述目标中心点的岩石类别，对所述特征向量所属的岩石类别进行判定。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

获得已上传网站系统的所述待识别岩石的图片；

对待识别岩石的图片进行特征抽取，包括：

在所述网站系统的岩石识别界面中，在接收到岩石识别业务的触发操作时，对待识别岩石的图片进行特征抽取。

3.根据权利要求1所述的方法，其特征在于，根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别，包括：

在任意岩石类别的概率大于第一预设阈值时，输出所述待识别岩石属于该任意岩石类别的概率；

在所有岩石类别的概率都小于第一预设阈值时，获取概率大于第二预设阈值的多个岩石类别；

在所述多个岩石类别属于同一个岩石类别的上位类别时，输出该同一个岩石类别的上位类别。

4.根据权利要求1所述的方法，其特征在于，所述方法还包括：

对获取的岩石图像标记标准岩石类别，得到岩石图像样本；

对所述岩石图像样本进行特征抽取，得到特征向量样本；

将所述特征向量样本输入预设有多个中心点的高维度空间，得到所述特征向量样本在所述高维度空间中的第二投射点；

计算所述第二投射点与所述高维度空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；

根据所述第二投射点与所述高维度空间中预设的多个中心点的距离，确定与所述特征向量样本距离最近的预测中心点；

在所述预测中心点的岩石类别与所述标准岩石类别不一致时，调整所述高维度空间的参数，将经过多次参数调整的所述高维度空间作为所述预设岩石分类空间。

5.根据权利要求4所述的方法，其特征在于，调整所述高维度空间的参数，包括：

确定所述标准岩石类别对应的目标中心点；

按照(1)式计算所述第二投射点与所述目标中心点的损失值；

(1)式中，m为所述岩石图像样本的数量；x_i为所述第二投射点的位置坐标；c_yi为所述目标中心点的位置坐标；

根据所述损失值，调整所述预设高维度空间的参数。