摘要
本发明提供了一种焊接图像缺陷识别方法、装置、存储介质及设备;其中方法包括:获取待识别的焊接图像;将待识别的焊接图像输入至缺陷识别模型,通过所述缺陷识别模型对所述焊接图像进行识别,得到焊接图像类型,从而判定焊接图像是否存在缺陷以及存在缺陷时的缺陷类型;对初始缺陷识别模型进行训练处理为:利用预训练模型的卷积基结合全连接分类器来构成缺陷识别模型;对缺陷识别模型进行迁移训练;迁移训练是指冻结卷积基,训练全连接分类器,之后对缺陷识别模型进行微调。本发明基于深度学习和迁移学习技术结合,可在样本数量有限情况下实现缺陷识别模型的高效训练,提高焊接图像缺陷识别的准确率。
- 单位