采用Resnet18网络对美国西储大学轴承故障数据集进行诊断。
本次实验的数据来自美国西储大学轴承故障诊断数据集,48k/0HP,总共10个故障类别。
采用MATLAB对数据进行预处理;网络训练环境为python3.7 与 pytorch1.8.0,既可以用CPU也可以用GPU。
file1_data_process.m文件是对原始数据按7:2:1的比例进行划分,并保存在data_process.mat中。
file2_fft_preprocess.m是对划分后的data_process.mat进行FFT变换操作,并将处理后的数据保存在FFT.mat文件中。
“残差收缩网络.py”即为所提模型。
以下为实验结果(CPU跑的):