基于泊松融合数据增强的焊缝金相组织缺陷分类研究

基于焊缝金相组织图像对焊缝内部缺陷进行分类,是工业焊接质量检测的重要一环.为提高小样本(样本数不大于30)焊缝金相组织图像中缺陷的分类效果,采用泊松融合方法对缺陷图像进行数据增强,提出ResNet18_PRO分类网络模型,显著提升缺陷分类精度.数据增强方面,通过数字图像处理的方法提取出原缺陷样本中的缺陷区域,后利用泊松融合方法将缺陷区域与正常样本进行融合从而生成新的缺陷样本,以此扩充缺陷样本数据;网络模型方面,在ResNet18网络模型基础上,对其下采样结构进行改进,以减少原下采样结构带来的信息损失,同时在网络末端增加改进的空间金字塔池化(ISPP)结构,以整合多尺度的特征信息.通过多个分类模型对样本扩充前后的缺陷分类效果进行对比,验证了该数据增强方法对分类效果的提升具有较为显著的作用,同时对ResNet18_PRO网络模型进行消融实验,验证了网络各改进部分及训练策略改进的有效性.ResNet18_PRO模型对增强后的数据平均分类准确度达到98.83%,平均F 1分数达到98.76%,显著提高了金相组织缺陷的分类效果,将该模型运用于其他工业缺陷数据集取得了较好的分类效果.实验结果表明,该模型有良好的鲁棒性,具有较好的实用价值.