基于ViT框架的多类目岩矿快速分类方法、设备及存储介质

摘要

本发明提出了一种基于ViT框架的多类目岩矿快速分类方法、设备及存储介质，通过构建自然岩矿图像识别模型，其中自然岩矿图像识别模型的预处理模块对岩矿图像数据进行特征预提取和抑制过拟合处理，并输入至数据处理模块中进行参数训练和Token提取，再输入至结果处理模块根据所述参数训练和Token提取结果进行Token分离和类别映射并输出分类结果。本发明在构建自然岩矿图像识别模型时，在采用迁移卷积联合ViT框架的基础上，加入了STFE模块、ECTG模块、Mlti‑Head P‑Attention模块以及RCLinaer层，从而减缓过拟合，提升映射的容错率，更加精确完整地提取图像数据的特征，实现了减少岩矿图像数据中干扰因素对分类结果的影响，在大容量多类目的情况下实现对岩矿图像的快速精准识别的目的。