基于改进残差网络与迁移学习的铜合金金相图分类方法
Classification method of copper alloy metallographic diagram based on improved residual network and transfer learning
-
摘要: 针对铜合金领域特定金相的数据集通常非常小,无法满足传统卷积神经网络建模需要大量训练样本的问题,提出一种基于改进残差网络与迁移学习的铜合金金相图分类方法,即基于ImageNet数据集和金相图训练集预先训练ResNet50残差模型,训练时采用迁移学习(Transfer Learning)方法并重新建立全连接层对铜合金金相类别进行分类识别。本方法的准确率为97.2%,精确率为95.6%,召回率为97.3%,F1分数为96.4%,优于VGG19和基于迁移学习的MobileV2等方法。实验结果表明,采用迁移学习方法可以克服金相图像数据集小的问题,使用ResNet50进行特征提取可以很好地获得铜合金金相图的纹理信息。本研究建立了一种铜合金金相结构自动分类和识别的新方法,可较准确地分类和识别铜合金金相图。Abstract: A copper alloy gold phase diagram classification method based on an improved residual network and transfer learning was proposed for the specific metallographic datasets in the copper alloy field, which are usually very small and cannot meet the problem of large training samples required for traditional convolutional neural network modeling. Based on ImageNet, the ResNet50 residual model was pre trained using the ImageNet dataset and metallographic diagram training set. Transfer learning method was used during training, and a fully connected layer was re-established to classify and recognize copper alloy metallographic categories. The experimental results show that the accuracy of this method is 97.2%, the precision is 95.6%, the recall is 97.3% and the F1-score is 96.4%, which is superior to VGG19 and MobileV2 based on transfer learning. Moreover, the transfer learning method can overcome the problem of small metallographic image datasets, and the use of ResNet50 for feature extraction can effectively obtain the texture information of copper alloy metallographic maps. This study establishes a new method for automatic classification and identification of metallographic structures of copper alloys, which can more accurately classify and identify the metallographic structures.