一种基于多尺度卷积神经网络的岩石岩性分类识别方法

    公开(公告)号:CN118587702A

    公开(公告)日:2024-09-03

    申请号:CN202410585628.6

    申请日:2024-05-11

    摘要: 本发明公开了一种基于多尺度卷积神经网络的岩石岩性分类识别方法,对岩石薄片显微图像进行数据增强,得到数据增强后的岩石薄片显微图像数据集,按比例将岩石薄片显微图像数据集划分为训练数据集和测试数据集,通过训练数据集和测试数据集对改进的MobileViT模型进行训练和验证,最终得到训练好的岩石薄片显微图像分类模型,对岩石类型的名称和置信度进行预测。通过本发明可以快速准确地获得岩石薄片显微图像的类别。