1.一种基于改进WGAN网络的轴承故障诊断方法，其特征在于，包括如下步骤：

（1）加速度传感器的选取与安装，

（2）采集样本轴承运行过程中的振动信号并对样本信号进行处理，

（3）构建改进WGAN诊断模型，利用该模型实现轴承故障诊断和故障分类，具体步骤为：（3.1）搭建生成器模型并生成虚假信号，

（3.2）采用R-FCN网络模型代替原来鉴别器模型，获取改进WGAN诊断模型，（3.3）Hyperopt优化改进WGAN诊断模型超参数，

（3.4）采用半监督学习指导改进WGAN诊断模型中生成器和鉴别器训练，

（3.5）优化后改进的WGAN诊断模型实现轴承故障诊断，

（4）采集待测轴承运行的全生命周期振动信号并对振动信号进行处理，将处理后的振动信号以时频图形式输入改进WGAN诊断模型，以获取待测轴承的诊断精度和故障类型。

2.根据权利要求1所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（1）中将加速度传感器安装在齿轮箱的驱动端，将加速度传感器安装在齿轮箱驱动端表面9时和12时方向，且加速度传感器安装位置相对垂直。

3.根据权利要求2所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（2）的具体步骤为：（2.1）设置采样参数，采样参数包括采样频率、采样时长t=t'和采样间隔，对轴承所受加速度信号进行采样，采集轴承运行过程中的样本数据；

其中，样本数据由人为故障注入获取，主要包括正常轴承和故障轴承，故障轴承包括6种不同类型故障的轴承；7种轴承分配情况为：样本轴承1为损伤直径为的正常轴承，振幅为；样本轴承2为外圈损伤直径为的故障轴承，振幅为；样本轴承3为内圈损伤直径为的故障轴承，振幅为；样本轴承4为滚珠体损伤直径为的故障轴承，振幅为；样本轴承5为外圈损伤直径为的故障轴承，振幅为；样本轴承6为内圈损伤直径为的故障轴承，振幅为；样本轴承7为滚珠体损伤直径为的故障轴承，振幅为；7种不同数据又将轴承分为不同故障程度，其中，为正常数据、为轻微故障数据、为中度故障数据、以上为重度故障数据；

（2.2）将步骤（2.1）中采样结果以数值形式输出并保存，轴承在高速运转时，驱动端的加速度传感器将采集到加速度振动信号，采集到加速度振动信号通过模数转换器进行转换，转换后的加速度振动信号按照桥式电路放大输出振动信号值；最后将每次采样输出的数值以CSV格式保存，分别对两种不同方向的采样点进行采样，9时方向加速度振动信号放置在CSV文件第一列，12时方向加速度振动信号放置在CSV文件第二列，保存为1.CSV，以此类推，完成数据保存；

（2.3）步骤（2.2）中数据进行分割处理，若一个CSV文件中有N个数据，取前N⁄1000个数据为一组，以此类推，逐个遍历CSV文件，共得到组数据；

（2.4）对步骤（2.3）中获取的组数据进行连续小波变换处理，得到相对应个可作为数据集的可视化时频图，时频图为RGB格式且像素为m*n；

（2.5）采用独热编码对步骤（2.4）中数据集按照训练集和测试集分别为80%和20%进行标签处理，利用0和1表示运行数据的故障状态，即使用N位寄存状态器N种状态进行编码，设置1000为正常状态，0100为轻微故障状态，0010为中度故障状态，0001为重度故障状态；并利用反独热编码表示输出类型，其中输出值为0代表正常状态，输出值为1代表轻微故障状态，输出值为2代表中度故障状态，输出值为3代表重度故障状态。

4.根据权利要求3所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.1）中搭建生成器网络模型的具体步骤如下：（3.1.1）采用二维卷积层作为网络输入，输入端为随机噪声数据，经批量标准化处理实现数据归一化，再采用LeakyRule激活函数激活大量神经元，并依靠神经元间的计算提取深层次特征，最后利用tanh激活函数作为网络输出；

（3.1.2）将设置的具有独热编码标签的随机噪声输入至生成器中，经过生成器自身网络结构训练生成器生成虚假信号；

（3.1.3）利用改进的CWT对随机噪声生成的虚假信号进行处理，以生成所需的时频图。

5.根据权利要求4所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.2）中搭建R-FCN网络模型为改进WGAN网络的鉴别器模型，利用鉴别器模型对步骤（3.1）中生成器生成图片进行判断，并输出生成器和鉴别器损失函数及改进WGAN模型的诊断精度，具体步骤如下：（3.2.1）利用O个二维卷积层Conv2D、P个Batch Normalizaztion层、Q个LeakyRule激活函数及ResNet101网络组成鉴别器输入端网络，在通过1个Conv2D得到初始特征图，再对其进行Conv2D操作，最终得到去除冗余图像特征的新特征图；

（3.2.2）通过区域提议网络RPN获取新特征图中对应的感兴趣区域边缘框，并利用每个RoI边缘框分别获取步骤（3.2.1）中新特征图中的特征；再通过每个RoI边缘框区域的r，s，t，u来确定位置敏感区域的坐标信息，r，s，t，u为边缘框坐标位置，依据该方法得到每个子区域，最终由每个RoI实现特征的分类和回归；

其中，RPN网络是Faster R-CNN中的RPN层，主要用于Faster-RCNN网络提取边缘框；由于提取R-FCN网络中边缘框耗时耗力，因此，在引入卷积神经网络后，采用的二维卷积层以特征提取的形式选择边缘框的位置从而降低了计算时间，在此发明中RPN网络可较快的提取R-FCN网络中的边缘框；

（3.2.3）在每个子区域上利用池化操作寻找合适的响应值，依据RoI边缘框大小设置最小响应值为S，最大响应值为T，若响应值在[S，T]之间，则表示该特征为所需特征；反之，去除该特征，继续寻找合适特征直至找到所有特征；直到获取整张图片中的特征；

（3.2.4）将步骤（3.2.3）中每个RoI得到的特征按照获取边缘框的大小映射至新特征图，得到位置敏感区域，即通过池化操作得到响应值在[S，T]之间的特征区域，其中，得分映射大小，即位置敏感区域获取的边缘框大小，为K*K*(C+1)和4*K*K维，再利用卷积池化操作对新特征图上的位置敏感区域进行特征提取，并将提取结果经全连接层输出；

（3.2.5）将步骤（2.5）中带有标签的真实训练集输入至搭建好的鉴别器模型进行训练，再将步骤（3.1）中生成器生成图片输入至R-FCN中进行训练，利用鉴别器损失函数和生成器损失函数进行判断，如果鉴别器鉴别结果为真，同时生成器损失函数和鉴别器损失函数下降，当两者最终趋于拟合并接近时，达到纳什平衡，表明诊断模型满足要求，则训练结束；再利用softmax函数分类得到带有独热编码标签的故障类型，并输出随迭代次数变化生成器和鉴别器模型的损失函数曲线和改进WGAN网络模型诊断精度曲线；如果鉴别器判断结果为假，则进行步骤（3.2.6）；

（3.2.6）训练结果为假时，则表示生成器生成图片效果不好，鉴别器无法正确判断图片真假，生成器模型未达到预期效果，则采用控制变量法控制鉴别器模型及其超参数来优化生成器模型，将鉴别器的模型参数固定，不断迭代优化生成器的模型参数，当鉴别器判断结果为真时，则生成器模型达到最优，此时鉴别器和生成器模型优化效果最优，效果最优就是当生成器和鉴别器损失函数下降，且两者最终趋于拟合并接近时，停止迭代，以此获取改进WGAN诊断模型；其中在获取改进WGAN诊断模型时，其迭代次数为epochs，学习率为Lr，批量大小为Batch_size；

这里的控制变量法就是保持鉴别器模型和参数不变，去优化生成器；

（3.2.7）利用测试集对步骤（3.2.5）中获取的改进WGAN诊断模型进行测试；若鉴别器可以判断生成器生成图片真假，且鉴别器损失函数和生成器损失函数缓慢降低至拟合，不在出现陡坡式的跳跃性波动，即不存在梯度消失或崩塌现象，则停止迭代，输出生成器和鉴别器损失函数和模型训练精度，得到改进的WGAN诊断模型。

6.根据权利要求5所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.2）中为进一步提高改进WGAN诊断模型的诊断速度，对R-FCN网络中Resnet101网络进行层数优化，分别选取和五个层级，为101层，将每个RoI区域从划分为个格子，设置迭代次数为epochs-1，学习率为Lr-1，批量大小为Batch_size-1，在此基础上进行训练，训练结果以训练精度体现，从五个层级中得到最优层数为，相对于101层，Resnet网络减少了网络层数，提高了训练速度，在步骤（3.2）基础上得到去除冗余网络层的诊断模型。

7.根据权利要求6所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.3）由于改进WGAN网络模型仍存在较多网络结构和众多超参数，会影响生成器生成能力和鉴别器的鉴别能力，为实现深层次信号的特征提取，提高诊断精度，利用自动调参优化鉴别器模型达到优化改进WGAN模型目的，具体步骤如下：（3.3.1）初始化网络超参数，获取待优化的卷积核大小、隐藏神经元个数、池化因子参数组成初始参数集，选取第一组参数将其输入至改进WGAN模型进行训练；

其中，初始化参数集为：；

（3.3.2）建立目标函数，目标函数作为评估先验模型的基础，用于计算更新优化函数的后验概率，得到最优超参数组合；采用交叉熵损失函数作为目标函数，公式如下：（1）

式中，B为样本数，C为类别数，为第b个样本是否属于第c个类别，为分类的输出结果，为正则化系数，为网络层待学习参数，j为特征映射；

（3.3.3）将步骤（3.3.1）中的初始参数利用步骤（3.3.2）目标函数进行计算，得到对应的函数评估值；其中，，由和构建数据集；

（3.3.4）将步骤（3.3.3）数据集A中的第一组数据用于验证集上判断模型是否满足要求；具体可依据模型的超参数组合表达式判断，若鉴别器损失函数迭代趋于拟合，则该组超参数满足要求；反之，进行步骤（3.3.5），公式如下：（2）

式中，为最小化的目标函数；为获取的一组最优超参数；

设置最小目标函数时，满足超参数选取要求；

（3.3.5）利用数据集A建立高斯回归模型，通过高斯回归模型不断迭代计算更新损失函数，修正概率模型；

为在高斯分布中获取最优超参数，可利用极大似然估计法，其表达式如下：

（3）

式中，K为协方差矩阵；f为前n个样本的后验概率；为所选超参数，；

（3.3.6）使用采集函数获取参数集中下一组超参数，将其输入至步骤（3.3.3）计算得到新的评估值；

其中，步骤（3.3.6）中采集函数是由当前数据集的后验分布构造，通过最大化采集函数来选取下一组超参数，采集函数公式如下：（4）

式中，为训练集，、分别为目标函数联合后验分布的均值函数和协方差函数；为调节参数，可用来调整采样点的选取；

（3.3.7）判断是否满足模型精度要求，若满足，则确定超参数及；反之，则继续执行步骤（3.3.2）至步骤（3.3.6），直至满足要求，停止迭代；

得到改进WGAN模型的最优网络超参数，进而实现Hyperopt优化改进WGAN诊断模型的目的，再利用步骤（3.2.6）实现生成器和鉴别器双重优化，获取优化后的改进WGAN诊断模型。

8.根据权利要求7所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.4）中为解决大多数WGAN网络无监督学习模型不仅需要大量样本数据，还存在样本标注困难问题；通过更改鉴别器和生成器损失函数，实现半监督学习，目的在于使WGAN网络学习器不依赖外界交互、自动利用未标记样本来提升学习性能，具体步骤如下：（3.4.1）构建半监督学习中鉴别器的损失函数，

首先，构建WGAN网络的监督学习损失函数，公式如下：

（5）

式中，E为数学期望，真实数据和的概率分布，是鉴别器判断为假的概率；

其次，构建WGAN网络的无监督学习损失函数，公式如下：

（6）

式中，为判别器对真实数据的评估，为生成数据，为随机噪声，为真实数据，为鉴别器对生成数据的评估，为虚假数据；

最后，构建改进WGAN网络模型的鉴别器损失函数，公式如下：

（7）

进一步地，为提高鉴别器损失函数精确度，将监督学习和半监督学习损失函数权值归一化处理并乘以权值尺度因子，则更新后的损失函数表达式如下：（8）

（3.4.2）改进WGAN网络半监督学习中生成器损失函数的构建，公式如下：

（9）

（3.4.3）利用步骤（3.3）中优化好的网络模型为基础，训练步骤（3.1）中生成的具有独热编码标签的生成图片和用样本标签率代替整体训练样本标签的真实图片，样本标签率为四种，其中不为0，表示监督学习；

（3.4.4）对步骤（3.4.3）中的训练结果进行判断，通过不同的样本标签率进行训练分别得到相应准确率；当为时，对应得最高，此时，半监督学习较监督学习和无监督学习效率更高，再利用softmax函数分类得到带有独热编码标签的故障类型，并输出随迭代次数变化的损失函数曲线和网络模型精度损失曲线，完成训练。

9.根据权利要求8所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（3.5）中具体改进的WGAN整体诊断流程如下：（3.5.1）将随机噪声和步骤（3.4）中具有独热编码的样本标签率输入步骤（3.3）优化后的生成器模型中，经过生成器自身网络结构训练生成虚假信号；

其中，随机噪声需满足格式且像素与时频图大小相同；

（3.5.2）利用上述改进的CWT对信号进行连续小波变换使其得到标准的时频图，使其更加具备‘欺骗’鉴别器能力；

（3.5.3）将步骤（3.5.2）得到的时频图输入至步骤（3.3）优化后的R-FCN鉴别器网络模型，同时将训练集输入至鉴别器中，经鉴别器迭代训练，判断步骤（3.5.2）输入的时频图是否为真，如果为真，则鉴别器和生成器损失函数达到拟合效果，输出随迭代次数变化生成器和鉴别器的损失函数曲线、改进WGAN网络模型的诊断精度曲线及独热编码标签的故障类型；反之，固定鉴别器参数，不断迭代更新生成器参数，直至生成器生成数据鉴别器无法判断是真是假，最终得到改进WGAN网络模型，再将测试集输入改进WGAN网络模型中进行验证；

（3.5.4）通过Hyperopt优化方法得到诊断模型最优超参数组合，进而优化改进WGAN模型；

（3.5.5）为验证半监督学习是否可以在优化后的改进WGAN网络模型中实现故障诊断，利用不同样本标签率代替整体样本标签输入改进WGAN网络模型进行训练，并输出不同样本标签率的诊断精度。

10.根据权利要求9所述的一种基于改进WGAN网络的轴承故障诊断方法，其特征在于：所述步骤（4）的具体步骤如下：（4.1）获取轴承全生命周期运行数据，并将数据以CSV格式保存；

（4.2）将步骤（4.1）中CSV格式的数据通过频谱信号分析获取时间-振幅关系图，并按照振幅对数据进行划分，若振幅在之间，为正常状况；振幅在之间，为轻微故障；振幅在之间，为中度故障；振幅大于时，为重度故障；进而将划分为不同故障程度的数据集保存在Excel中；

（4.3）将步骤（4.2）Excel中的数据集按照80%和20%划分为训练集和测试集，并用步骤（2.5）中的标签处理方式对数据集做标签处理；

（4.4）将步骤（4.3）中处理后的数据输入Hyperopt优化后的改进WGAN网络模型中训练，输出随迭代次数变化生成器和鉴别器的损失函数曲线、改进WGAN网络模型的诊断精度曲线及独热编码标签的故障类型。