基于改进ResNet两阶段训练模型的转子焊锡缺陷检测  

Rotor Solder Defect Detection Based on Improved ResNet Two Stage Training Model

作　　者：吴承俊 方夏 王杰[1] 张帆 李秦川 

机构地区：[1]四川大学机械工程学院,四川成都610065

出　　处：《工业控制计算机》2023年第4期115-116,119,共3页Industrial Control Computer

基　　金：四川省重点科技研发项目(2021YFG0198)。

摘　　要：针对当前转子焊锡缺陷检测准确率较低的问题,提出了一种基于改进ResNet的两阶段训练网络模型。首先在主干网络引入SRM注意力机制,提高网络对色泽纹理的特征提取能力,通过改进的相似度损失函数训练特征编码器,然后再通过添加分类头的方法进行微调训练出最终的网络模型。将提出的方法用于转子焊锡缺陷检测,并与经典的ResNet网络等比较,实验证明,采用所提出的方法准确率可达到97.6%,明显优于经典的ResNet等分类方法,具有一定的应用价值。Aiming at the low accuracy of the current rotor solder defect detection,a two-stage training network model based on improved ResNet is proposed.Firstly,SRM attention mechanism is introduced into the backbone network to improve the network's ability to extract features of texture and shape.The feature encoder is trained through improved similarity loss,and then the final network model is finely tuned by adding classification heads.The proposed method is applied to rotor solder defect detection,and compared with the classical ResNet network,the experimental results show that the accuracy of the proposed method can reach 97.6%,which is significantly better than the classical ResNet classification methods.It has certain application value.

关 键 词：缺陷检测 相似度损失 转子焊锡 ResNet 

分 类 号：TP18[自动化与计算机技术—控制理论与控制工程] TG441.7[自动化与计算机技术—控制科学与工程]