岩石薄片识别方法、装置、设备及存储介质

本申请提供一种岩石薄片识别方法、装置、设备及存储介质。涉及图像处理技术领域。该方法包括：构建第一数据集和第二数据集；其中，第一数据集和第二数据集均包括有不同类别的岩石薄片图像；利用第一数据集对神经网络模型进行训练，得到特征权重；构建识别模型，识别模型包括依次连接的卷积层、展平层、多头自注意力层、全连接层、SoftMax层以及分类层；其中，卷积层的权重为特征权重；利用第二数据集对所述识别模型进行训练，得到训练后的识别模型，所述训练后的识别模型用于识别岩石薄片图像中的岩石薄片类别。本申请可以提升岩石薄片图像精细分类与预测的精确性。