transformer 滚轴故障诊断

transformer 是一种常用的神经网络模型，它在滚轴故障诊断中也有应用。根据引用所述，特征提取部分使用了matlab时频图变换工具箱和pytorch1.6实现的vision transformer（VIT）分类。引用中提到，传统轴承故障诊断通常采用各种特征提取方法对一维轴承信号进行处理，而本文使用小波时频图将一维轴承信号转换为二维的时频图，并以此作为VIT模型的输入进行故障诊断分类。据测试结果显示，使用这种方法进行滚轴故障诊断的精度可以达到99.5%。引用中提到了所用的数据集和数据划分方法。