基于生成对抗网络的轴承故障数据剧增的方法

本发明提供一种基于生成对抗网络的轴承故障数据剧增的方法，该方法包括：获取原始目标故障数据集，并输入至生成对抗网络模型，得到生成目标故障数据集；将生成目标故障数据集和原始目标故障数据集的包络频谱在故障特征频率下的幅值误差和频率误差的和作为第一修正项；将生成目标故障数据集加入至原始目标故障数据集，得到更新后的目标故障数据集后输入至故障分类模型，得到分类误差，并将分类误差作为第二修正项；根据第一修正项和第二修正项，对生成对抗网络模型的生成器的代价函数进行修正，生成修正后的生成对抗网络模型；将更新后的目标故障数据集输入到修正后的生成对抗网络模型，实现故障数据的剧增并平衡各个故障标签下的样本数量。