Rock Image Intelligent Classification and Recognition Based on Resnet-50 Model

Abstract

Machine learning algorithms becomes popular for intelligent classification of rock images. In this paper, it selects Resnet 50 neural network model to divide the data sets based on the rock pictures taken under the white light lamp. By continuously adjusting the parameters of each layer, the intelligent classification of rocks is carries out. The training final validates accuracy reached 94.12%.