故障诊断数据集及实现代码

文件包括凯斯西储大学轴承故障数据，分别在负载1.2.3下的十种故障类型，通过卷积神经网络方法对其进行故障诊断，准确率高达99.67%，使用python语言对其进行复现，里面包括具体对应的论文。亲测好用

# -*- coding: utf-8 -*-
“““
Created on Mon Jul  1 11:15:44 2019

@author: Administrator
“““

import os
import glob
import numpy as np
from scipy.io import loadmat
from Preprocessor import Preprocessor

class DataReader:
    
    def __init__（self 
                 data_augment_stride = 64
                 sample_size = 2048
                 test_set_num = 25
                 train_set_num = 660
                 valid_set_num = 7）:
        
        self.__data_augment_stride = data_augment_stride
        self.__sample_size = sample_size
        self.__test_set_num = test_set_num
        self.__train_set_num = train_set_num
        self.__valid_set_num = valid_set_num

    def read_data（self 
                  path 
                  sensor_place = ‘DE‘）:
        #所需数据长度，从最后一个开始，往回追溯
        length = （（self.__test_set_num + self.__valid_set_num） * self.__sample_size + 
                  self.__sample_size + （self.__train_set_num - 1） * self.__data_augment_stride）
        
        cate = [path + ‘\\‘ + x for x in os.listdir（path） if 
                os.path.isdir（path + ‘\\‘ + x）]
        
        train_data = np.zeros（（1 self.__sample_size））
        test_data = np.zeros（（1 self.__sample_size））
        valid_data = np.zeros（（1 self.__sample_size））
        train_label = np.zeros（（1））
        test_label = np.zeros（（1））
        valid_label = np.zeros（（1））
        
        for idx folder in enumerate（cate）:
            #读某一类数据一类数据仅一个文件
            for t in glob.glob（folder + ‘\*.mat‘）:
                print（‘reading the data:%s‘%（t））
                mat = loadmat（t）
                key = [key for key in mat.keys（） if sensor_place in key]
                data_temp = mat[key[0]][-length:]
            
            train_data_length = （self.__sample_size +
                                 （self.__train_set_num - 1） * self.__data_augment_stride）
            
            train_data = np.append（train_data 
                                   self.__data_segmentation（data_temp[: train_data_length]）
                                   axis = 0）
            
            test_data_length = self.__sample_size * self.__test_set_num
            
            test_data = np.append（test_data
                                  self.__data_segmentation（
                                          data_temp[train_data_length: 
                                              train_data_length + test_data_length] 
                                              ‘test‘）
                                    axis = 0）
                                      
            valid_data = np.append（valid_data 
                                   self.__data_segmentation（
                                           data_temp[train_data_length + test_data_length:] 
                                           ‘valid‘）
                                    axis = 0）
                                   
            train_label = np.append（train_label 
                                  

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\.idea\
     目录           0  2020-08-04 20:38  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\.idea\inspectionProfiles\
     文件         310  2020-08-04 20:38  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\.idea\misc.xml
     文件         489  2020-08-04 20:37  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\.idea\modules.xml
     文件       16137  2020-08-12 20:40  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\.idea\workspace.xml
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\__pycache__\
     文件        3642  2020-08-11 16:13  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\__pycache__\DataReader.cpython-36.pyc
     文件         809  2020-08-04 15:04  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\__pycache__\Preprocessor.cpython-36.pyc
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\0\
     文件     7742720  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\0\Normal.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\1\
     文件     2914248  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\1\0.007-Ball.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\2\
     文件     2931672  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\2\0.014-Ball.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\3\
     文件     2921232  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\3\0.021-Ball.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\4\
     文件     2928192  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\4\0.007-InnerRace.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\5\
     文件     2924712  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\5\0.014-InnerRace.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\6\
     文件     2917752  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\6\0.021-InnerRace.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\7\
     文件     2938632  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\7\0.007-OuterRace6.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\8\
     文件     2931672  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\8\0.014-OuterRace6.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\9\
     文件     2928192  2019-07-13 03:36  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\A\9\0.021-OuterRace6.mat
     目录           0  2020-08-12 22:23  Rolling_Element_Bearings_Fault_Intelligent_Diagnosis_based_on_Convolutional_Neural_Network_Using_Raw_Sensing_Signal\data\B\
............此处省略60个文件信息