Convolutional Neural Network Model Combined with Transfer Learning and Ensemble Learning for Rock Identification
A convolutional neural network model named TF_MDR_Fusion is proposed for rock identification.The model is combined with transfer learning and ensemble learning methods.It primarily embraces the concept of transfer learning,in-volving pretraining the MobileNetV3_Large,DenseNet121,and ResNet50 models on the ImageNet dataset.Three models are combined using ensemble learning techniques to further elevate the performance of rock identification.Through comparative experiments and ablation experiments,the results consistently demonstrate that TF_MDR_Fusion model outperforms individual models,with an accuracy of 77.60%in rock image identification.