格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)

 1.运行效果:

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码)_哔哩哔哩_bilibili

环境库

只要tensorflow版本大于等于2.4.0即可运行 

同样的模型应用于东南大学轴承数据集:格拉姆角场GAF将时序数据转换为图像并应用于东南大学轴承故障诊断(Python代码,CNN模型)_深度学习的奋斗者的博客-CSDN博客

2.GAF的内容

GAF是一种用于时间序列数据可视化和特征提取的技术,通常用于时间序列分类和分析。

GAF的基本思想是将时序数据转换为角度矩阵,然后将这个矩阵可视化为图像。这种可视化方式有助于人们更好地理解和分析时间序列数据。GAF的一种常见用法是在机器学习任务中,将时间序列数据转换为图像特征,然后使用这些特征进行分类、回归或聚类等任务。

3.数据集

3.1 西储大学轴承数据集(CWRU)

 CWRU数据集,每份负载数据集以四分类为例:正常、内圈故障、外圈故障和滚动体故障,也可以考虑故障尺寸,自己改成十分类)

自己也可以把数据集换成其他数据集或者自己的轴承数据集。

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第1张图片

 以0HP文件夹为例,打开0HP文件夹

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第2张图片

inner、Normal、outer和roller文件夹装载的原始数据集

picture_CNN.py主要功能:首先利用1024的固定长度重叠切割原始数据,生成一维样本,然后利用GAF将一维样本转变成二维图像的程序,然后利用CNN进行故障诊断。

 打开0HPimages文件格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第3张图片

随意选取一个样本照片展示 

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第4张图片

每种负载下5次实验平均结果 

0HP 1HP 2HP 3HP
99.58% 99.68% 99.58% 99.62%

0HP数据集结果(训练集与测试集比例:4:1)

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第5张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第6张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第7张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第8张图片

测试集混淆矩阵 (测试集准确率99.58%)

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第9张图片

1HP数据集结果(训练集与测试集比例:4:1)

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第10张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第11张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第12张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第13张图片

测试集混淆矩阵  (测试集准确率99.68%)

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第14张图片

2HP数据集结果(训练集与测试集比例:4:1) 

 格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第15张图片

 格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第16张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第17张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第18张图片

测试集混淆矩阵  

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第19张图片

3HP数据集结果(训练集与测试集比例:4:1) 

测试集混淆矩阵  

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第20张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第21张图片

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第22张图片

 格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第23张图片

 测试集混淆矩阵  

格拉姆角场GAF将时序数据转换为图像并应用于凯斯西储大学轴承故障诊断(Python代码,CNN模型)_第24张图片

对项目感兴趣的,可以关注最后一行 

from scipy.io import loadmat
import matplotlib.pyplot as plt
from sklearn.preprocessing import MinMaxScaler
import numpy as np
import tensorflow as tf
import pickle
import numpy as np
import matplotlib.pyplot as plt
import scipy.signal as scisig
from mpl_toolkits.axes_grid1 import make_axes_locatable
from pyts.datasets import load_gunpoint
#代码和数据集的压缩包:https://mbd.pub/o/bread/mbd-ZJ6bmJdu

你可能感兴趣的:(诊断领域创新代码,python,开发语言)