故障诊断 | 基于小波时频图与Swin Transformer的轴承故障诊断方法（PyTorch）

文章概述

基于小波时频图与Swin Transformer的轴承故障诊断方法
针对用传统的故障诊断方法难以对非线性非平稳的柴油机故障信号进行准确高效诊断的问题, 提出基于小波时频图与Swin Transformer的故障诊断方法。该方法可以有效结合小波时频分析在处理非线性非平稳信号方面的优势和Swin Transformer强大的图像分类能力, 通过连续小波变换将原始信号表示为小波时频图, 将小波时频图作为特征图输入到Swin Transformer进行训练, 实现故障状态识别。

程序设计

• cwt_swinTransformer

import  torch
import  os, glob
import  random, csv
import</