一种基于NAS的岩石岩性识别方法、终端设备及存储介质

摘要

本发明涉及一种基于NAS的岩石岩性识别方法、终端设备及存储介质，该方法中包括：S1：采集岩石图片并对其岩性进行标注，根据标注后的岩石图片构建训练集；S2：构建分类模型，分类模型采用基于NAS的网络模型作为主干网络，之后依次串联全卷积网络和分类网络；S3：通过ImageNet图像数据集对分类模型进行预训练后，采用迁移学习算法，在分类模型中载入预训练后得到的权重参数后，再通过训练集对分类模型进行训练，训练后得到最终分类模型；S4：通过最终分类模型对待识别的岩石图片中的岩性进行识别。本发明相比于传统的机器视觉方法具有良好鲁棒性和泛化能力。