Rock thin section identification is an important method in geological research. Traditional identification methods relying on polarized light microscopy has obvious shortcomings such as strong subjectivity, high cost and low efficiency. This paper proposes a rock thin plate recognition method considering multifeature fusion based on vision transformer. Vision transformer is combined with color, Histogram of Oriented Gradients and brightness features extracted manually, and the ability of global information capture is enhanced by feature stitching and transformer self-attention mechanism, aiming to overcome the limitations of traditional methods for complex rock features. The experimental results show that the accuracy of the model is 87.8%, the weighted accuracy, harmonic mean of precision and recall are 0.88, 0.87 and 0.87, respectively. Compared with the random forest model, which has higher speed and interpretability than the convolutional neural network model, the accuracy is significantly improved, which verifies the effectiveness of feature fusion and vision transformer in rock classification.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Rock Thin Plate Recognition Method Considering Multi-Feature Fusion Based on Vision Transformer

  • Xiran Li,
  • Yuhao Li,
  • Dan Jing

摘要

Rock thin section identification is an important method in geological research. Traditional identification methods relying on polarized light microscopy has obvious shortcomings such as strong subjectivity, high cost and low efficiency. This paper proposes a rock thin plate recognition method considering multifeature fusion based on vision transformer. Vision transformer is combined with color, Histogram of Oriented Gradients and brightness features extracted manually, and the ability of global information capture is enhanced by feature stitching and transformer self-attention mechanism, aiming to overcome the limitations of traditional methods for complex rock features. The experimental results show that the accuracy of the model is 87.8%, the weighted accuracy, harmonic mean of precision and recall are 0.88, 0.87 and 0.87, respectively. Compared with the random forest model, which has higher speed and interpretability than the convolutional neural network model, the accuracy is significantly improved, which verifies the effectiveness of feature fusion and vision transformer in rock classification.