公开(公告)号：CN117330314A

公开(公告)日：2024-01-02

申请号：CN202311263852.5

申请日：2023-09-27

Applicant: 大连海事大学

Inventor： 向川 ,  李伟晨 ,  魏亮亮 ,  韩冰 ,  史鹏飞

IPC: G01M13/045 ,  G06F18/214 ,  G06F18/10 ,  G06V10/774 ,  G06V10/26 ,  G06V10/40 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0455 ,  G06N3/08

Abstract: 本发明提供一种基于二维灰度图和Swin‑Transformer的轴承故障诊断方法，包括：将原始一维轴承振动信号经过矩阵变换，转换成二维灰度图像，作为轴承故障诊断的训练集、验证集和测试集；构建基于Swin‑Transformer的轴承故障诊断模型；将含有标签的训练集和验证集依次输入数据预处理模块和特征提取网络，获得二维灰度图的空间特征图，将空间特征图输入至分类网络，利用损失函数反向更新特征提取网络中的权重系数，直至达到预设迭代次数，由此形成训练好的轴承故障诊断模型；将测试集输入到训练好的轴承故障诊断模型中，实现轴承故障类型识别。本发明主要通过矩阵变换处理一维轴承振动信号，得到二维灰度图，采用Swin‑Transformer网络来进行故障诊断。