In the development of high-tech industries, graphite has become increasingly more important. The world has gradually entered the graphite era from the silicon era. In order to make good use of high-quality graphite resources, a graphite classification and recognition algorithm based on an improved convolution neural network is proposed in this paper. Based on the self-built initial data set, the offline expansion and online enhancement of the data set can effectively expand the data set and reduce the risk of deep convolution neural network overfitting. Based on the visual geometry group 16 (VGG16), residual net 34 (ResNet34), and mobile net Vision 2 (MobileNet V2), a new output module is redesigned and loaded into the full connection layer. The improved migration network enhances the generalization ability and robustness of the model; moreover, combined with the focal loss function, the superparameters of the model are modified and trained on the basis of the graphite data set. The simulation results illustrate that the recognition accuracy of the proposed method is significantly improved, the convergence speed is accelerated, and the model is more stable, which proves the feasibility and effectiveness of the proposed method.
Graphite Classification Based on Improved Convolution Neural Network