基于深度学习的稀有金属花岗岩岩性细粒度分类研究

Research on fine-grained classification of rare metal granite lithology based on deep learning

  • 摘要: 细粒度图像分类在实际应用中具有较高的研究价值和应用前景。目前,传统岩性细粒度分类方法多依赖研究者的经验与实验设备的质量,主观性强且时效性差。文中将细粒度图像分类技术引入花岗岩岩性识别领域,系统构建了肉红色、灰白色、含铁锰质和含天河石碱长花岗岩4类岩性的RGB图像数据集,并采用AlexNet、VGG16、ResNet50和Vision Transformer等典型网络模型开展对比试验。结果表明:所有模型分类精度均超过82 %,其中VGG16模型最优,分类精度为88.57 %,较AlexNet模型提升了5.71 %;含天河石碱长花岗岩因特征矿物显著,识别准确率达100 %,而灰白色碱长花岗岩识别效果最差;模型精度与训练样本量呈正相关,完整训练集时模型性能最优。未来, 可通过改善岩石样本的数量和质量、优化模型算法,进一步提升稀有金属花岗岩岩性细粒度分类的精度。

     

    Abstract: Fine-grained image classification has high research value and application prospects in practical applications. At present, the traditional lithology fine-grained classification method is highly subjective and time-sensitive, depending on the experience of researcher and the quality of experimental equipment. Therefore, in this paper, the technology of fine-grained image classification is introduced into the field of granite lithology identification. The RGB image datasets of four types of lithology, namely flesh-red, grayish-white, iron-manganese, and amazonite-bearing alkali feldspar granites, are systematically constructed. Comparative experiments were carried out using typical deep learning models such as AlexNet, VGG16, ResNet50 and Vision Transformer. The results show that the classification accuracy of all models exceeds 82 %, and the VGG16 model is the best, which is 88.57 %, an increase of 5.71 % over the AlexNet model; the amazonite-bearing alkali feldspar granite has a recognition accuracy of 100 % due to its significant characteristic minerals, while the grayish-white alkali-feldspar granite has the worst recognition effect; the model accuracy is positively correlated with the amount of training samples, and the model performance is optimal when the training set is complete. In the future, the accuracy of fine-grained classification of rare metal granite lithology can be further improved by improving the quantity and quality of rock samples and optimizing the model algorithm.

     

/

返回文章
返回