基于单通道图数据增强和迁移训练残差网络的五相异步电动机滚动轴承故障诊断方法
摘要:
本发明是一种基于单通道图数据增强和迁移训练残差网络的五相异步电动机滚动轴承故障诊断方法,适用于非平衡小样本情况,属诊断技术领域。该方法将样本划分为训练集与测试集,使用GASF对训练集信号编码而得到单通道图;再将单通道图作为样本训练WGAN‑GP网络,得到样本对应的生成器网络;之后,根据迁移训练的原理,利用ImageNet预训练的残差网络卷积层作为特征提取层构造残差分类网络。进而,进行训练:使用生成器网络生成的样本对全连接层进行训练;使用实际样本作为训练集依次训练全连接层和特征提取层;对整个残差网络进行微调。将训练好的模型存储,调用即可完成诊断,准确率不低于99.4%。
0/0