一种基于迁移学习的金属断口图像语义分割方法及装置

本发明涉及一种基于迁移学习的金属断口图像语义分割方法及装置，本发明收集不同断口类型的金属断口图像作为图像数据集；对图像数据集中的金属断口图像进行语义分割掩膜标注与数据增强处理；将各类断口的图像数据集预处理后输入到迁移到图像特征处理模型获取低分辨率特征；设置断口图像语义分割任务的训练策略；将低分辨率特征输入构建的语义分割神经网络进行训练以使得语义分割神经网络模型能够预测与语义分割掩膜标注匹配的分割结果；对迁移学习训练所得的深度神经网络模型进行评价。本发明能够以小规模数据输入和短周期的深度神经网络训练过程，实现高准确率的不同类型金属断口图像语义分割。