Automatic rock classification and recognition algorithm based on Swin-Transformer
In the field of geological exploration,accurate rock identification is crucial for resource assessment,exploration po-sitioning,and environmental protection.However,traditional rock identification methods rely on the observation and experience of geologists,which suffer from low efficiency,subjectivity,and dependence on expert knowledge.To overcome these challenges,we proposes an automatic rock classification and recognition algorithm based on Swin-Transformer.The algorithm introduces a staged attention mechanism,dividing the image into different blocks and using windowed attention to allow each block to interact only with its neighboring blocks,significantly reducing computational and memory costs.Experimental results demonstrate that compared to popular classification models such as ResNet and EfficientNet,the proposed Swin-Transformer model achieves a significant im-provement in classification efficiency and accuracy,with a Top-1 accuracy of 91.3%and a Top-5 accuracy of 98.56%.This re-search not only enhances the accuracy and efficiency of rock identification but also reduces the reliance on manual intervention by automating the process,providing robust support for geological research and engineering applications.
rock identificationSwin-Transformerclassification recognitionmachine learninggeological exploration
万方数据
维普
