本申请提供了一种岩石识别方法、装置、设备及存储介质，涉及岩石识别技术领域。可以利用便捷的网站作为工具对岩石进行识别，排除了个人主观因素的影响，给岩石爱好者在野外对岩石种类进行快速准确的识别提供了一种途径。对待识别岩石的图片进行特征抽取，得到特征向量；利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，得到中间分类特征向量；根据所述中间分类特征向量，计算所述待识别岩石属于不同岩石类别的概率，得到概率分布向量；根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别。岩石所属的岩石类别的上位类别。岩石所属的岩石类别的上位类别。
【技术实现步骤摘要】
一种岩石识别方法、装置、设备及存储介质
[0001]本申请涉及岩石识别
，特别是涉及一种岩石识别方法、装置、设备及存储介质。
技术介绍
[0002]岩石广泛的存在于大自然之中，具有建筑材料、装饰品、能源材料等用途和价值。普通大众因为对岩石的认知偏少，缺乏岩石识别的知识，不能识别岩石的类别，而鉴定专家可以根据岩石的颜色、矿物组分、结构构造等特征来识别岩石类型，但无法提供随时随地的岩石识别服务。岩石外观可以通过拍摄照片，获取岩石图像等方式呈现，因此需要设计一种岩石识别方法、装置、设备及存储介质来代替鉴定专家进行岩石识别。
技术实现思路
[0003]本申请实施例提供一种岩石识别方法、装置、设备及存储介质，本申请可以利用便捷的网站作为工具对岩石进行识别，排除了个人主观因素的影响，给岩石爱好者普通大众在野外对岩石种类进行快速准确的识别提供了一种途径。
[0004]本申请实施例提供一种岩石识别方法，所述方法包括：
[0005]对待识别岩石的图片进行特征抽取，得到特征向量；
[0006]利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，得到中间分类特征向量；
[0007]根据所述中间分类特征向量，计算所述待识别岩石属于不同岩石类别的概率，得到概率分布向量；
[0008]根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别。
[0009]可选地，所述方法还包括：
[0010]获得已上传网站的所述待识别岩石的图片；
[0011]对待识别岩石的图片进行特征抽取，包括：
[0012]在所述网站系统的岩石识别界面中，在接收到岩石识别业务的触发操作时，对待识别岩石的图片进行特征抽取。
[0013]可选地，根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别，包括：
[0014]在任意岩石类别的概率大于第一预设阈值时，输出所述待识别岩石属于该任意岩石类别的概率；
[0015]在所有岩石类别的概率都小于第一预设阈值时，获取概率大于第二预设阈值的多个岩石类别；
[0016]在所述多个岩石类别属于同一个岩石类别的上位类别时，输出该同一个岩石类别的上位类别。
[0017]可选地，利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，包括：
[0018]将所述特征向量映射到所述预设岩石分类空间中，得到所述特征向量在所述预设岩石分类空间中的第一投射点；
[0019]计算所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；
[0020]根据所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离，确定与所述特征向量距离最近的目标中心点；
[0021]根据所述目标中心点的岩石类别，对所述特征向量所属的岩石类别进行判定。
[0022]可选地，所述方法还包括：
[0023]对获取的岩石图像标记标准岩石类别，得到岩石图像样本；
[0024]对所述岩石图像样本进行特征抽取，得到特征向量样本；
[0025]将所述特征向量样本输入预设有多个中心点的高维度空间，得到所述特征向量样本在所述高维度空间中的第二投射点；
[0026]计算所述第二投射点与所述高维度空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；
[0027]根据所述第二投射点与所述高维度空间中预设的多个中心点的距离，确定与所述特征向量样本距离最近的预测中心点；
[0028]在所述预测中心点的岩石类别与所述标准岩石类别不一致时，调整所述高维度空间的参数，将经过多次参数调整的所述高维度空间作为所述预设岩石分类空间。
[0029]可选地，调整所述高维度空间的参数，包括：
[0030]确定所述标准岩石类别对应的目标中心点；
[0031]按照(1)式计算所述第二投射点与所述目标中心点的损失值；
[0032][0033](1)式中，m为所述岩石图像样本的数量；x
i
为所述第二投射点的位置坐标；c
yi
为所述目标中心点的位置坐标；
[0034]根据所述损失值，调整所述预设高维度空间的参数。
[0035]本申请实施例第二方面提供一种岩石识别装置，所述装置包括：
[0036]第一识别模块，用于对待识别岩石的图片进行特征抽取，得到特征向量；
[0037]判定模块，用于利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，得到中间分类特征向量；
[0038]计算模块，用于根据所述中间分类特征向量，计算所述待识别岩石属于不同岩石类别的概率，得到概率分布向量；
[0039]输出模块，用于根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别。
[0040]可选地，所述装置还包括：
[0041]获得模块，用于获得已上传网站系统的所述待识别岩石的图片；
[0042]所述第一识别模块包括：
[0043]识别子模块，用于在所述网站系统的岩石识别界面中，在接收到岩石识别业务的
触发操作时，对待识别岩石的图片进行特征抽取。
[0044]可选地，所述输出模块包括：
[0045]第一输出子模块，用于在任意岩石类别的概率大于第一预设阈值时，输出所述待识别岩石属于该任意岩石类别的概率；
[0046]获取子模块，用于在所有岩石类别的概率都小于第一预设阈值时，获取概率大于第二预设阈值的多个岩石类别；
[0047]第二输出子模块，用于在所述多个岩石类别属于同一个岩石类别的上位类别时，输出该同一个岩石类别的上位类别。
[0048]可选地，所述判定模块包括：
[0049]映射子模块，用于将所述特征向量映射到所述预设岩石分类空间中，得到所述特征向量在所述预设岩石分类空间中的第一投射点；
[0050]计算子模块，用于计算所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；
[0051]确定距离子模块，用于根据所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离，确定与所述特征向量距离最近的目标中心点；
[0052]判定子模块，用于根据所述目标中心点的岩石类别，对所述特征向量所属的岩石类别进行判定。
[0053]可选地，所述装置还包括：
[0054]标记模块，用于对获取的岩石图像标记标准岩石类别，得到岩石图像样本；
[0055]第二识别模块，用于对所述岩石图像样本进行特征抽取，得到特征向量样本；
[0056]输入模块，用于将所述特征向量样本输入预设有多个中心点的高维度空间，得到所述特征向量样本在所述高维度空间中的第二投射点；
[005本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种岩石识别方法，其特征在于，所述方法包括：对待识别岩石的图片进行特征抽取，得到特征向量；利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，得到中间分类特征向量；根据所述中间分类特征向量，计算所述待识别岩石属于不同岩石类别的概率，得到概率分布向量；根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别。2.根据权利要求1所述的方法，其特征在于，所述方法还包括：获得已上传网站系统的所述待识别岩石的图片；对待识别岩石的图片进行特征抽取，包括：在所述网站系统的岩石识别界面中，在接收到岩石识别业务的触发操作时，对待识别岩石的图片进行特征抽取。3.根据权利要求1所述的方法，其特征在于，根据所述概率分布向量，至少输出所述待识别岩石属于一类岩石类别的概率或所述待识别岩石所属的岩石类别的上位类别，包括：在任意岩石类别的概率大于第一预设阈值时，输出所述待识别岩石属于该任意岩石类别的概率；在所有岩石类别的概率都小于第一预设阈值时，获取概率大于第二预设阈值的多个岩石类别；在所述多个岩石类别属于同一个岩石类别的上位类别时，输出该同一个岩石类别的上位类别。4.根据权利要求1所述的方法，其特征在于，利用预设岩石分类空间对所述特征向量所属的岩石类别进行判定，包括：将所述特征向量映射到所述预设岩石分类空间中，得到所述特征向量在所述预设岩石分类空间中的第一投射点；计算所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离；其中，每个中心点与一类岩石类别对应；根据所述第一投射点与所述预设岩石分类空间中预设的多个中心点的距离，确定与所述特征向量距离最近的目标中心点；根据所述目标中心点的岩石类别，对所述特征向量所属的岩石类别进行判定。5.根据权利要求1所述的方法，其特征在于，所述方法还包括：对获取的岩石图像标记标准岩石类别，得到岩石图像样本；对所述岩石图像样本进行特征抽取，得到特征向量样本；将所述特征向量样本输入预设有多个中心点的高维度空间，得到所述特征向量样本在所...
【专利技术属性】
技术研发人员：白林，彭伟航，叶江，周仲礼，郭凌瑜，陈涛，吴珺泓，段枕贞，邓清佩，
申请(专利权)人：四川地脑科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。