Abstract:
Mineral identification is a critical task in geological research, yet accurately identifying minerals remains a significant challenge. This study proposes an intelligent mineral image recognition model based on an improved ConvNeXt network, which utilizes transfer learning strategies and incorporates channel attention mechanisms to address the morphological characteristics of minerals.Firstly, the ConvNeXt network model pre-trained on the ImageNet dataset is employed and integrated into the mineral recognition model through transfer learning. Secondly, based on the ConvNeXt network, the model enhances feature fusion capabilities by combining the ConvNeXt blocks with attention mechanisms. Finally, a dataset comprising 34576 ore images of 26 mineral categories is used, divided into training, validation, and test sets in a 6∶2∶2 ratio. During experimental training, the proposed model demonstrates a significantly faster convergence compared to VGG19, GoogLeNet, ResNet50, ResNeXt50, and the ConvNeXt networks.Experimental results indicate that the intelligent mineral recognition model achieves an accuracy, precision, and recall of 98.58%, 98.62%, and 98.73%, respectively. Ablation experiments confirm that the optimization methods proposed in this study enhance model performance. Additionally, comparative visual analysis of feature maps from different models substantiates that the proposed mineral recognition model accurately extracts mineral features, further validating the model's effectiveness and improving mineral identification accuracy.