基于改进EEMD和生成对抗网络的旋转机械故障诊断方法及系统

本发明提供基于改进EEMD和生成对抗网络的旋转机械故障诊断方法及系统，属于旋转机械故障诊断技术领域，获取振动信号，通过改进EEMD算法将振动信号分解为多个本征模态分量，形成一维振动信号；将一维振动信号通过灰度处理转化为二维图像，并构建单通道二维图像训练数据集；建立旋转机械故障诊断模型，并设定目标函数；基于鉴别器损失及生成器损失训练生成对抗网络，并扩充故障数据，利用真实训练数据及生成训练数据训练分类器实现故障诊断。本发明结合改进EEMD和生成对抗网络进行数据不平衡情况下故障诊断，可以对旋转机械关键部件在复杂工况下所训练的诊断模型具有较好的稳定性、鲁棒性。