#?-*-?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個文件信息