weiyuqing/ECG_by_Diffenert_CNN_Test
1
0 Fork 0
mirror of https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git synced 2026-10-17 19:05:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update ECG_read_by_wfdb.py

Browse Source
This commit is contained in:
魏遇卿 committed 2023-04-25 13:16:30 +08:00
1 parent e44ea0019c
commit 8d80de2fa3
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
some-test/ECG_read_by_wfdb.py
+2 -2
View File
@@ -5,11 +5,11 @@ import numpy as np
def getDataset(number, X_data):
# get ECG from mit-bih-arrhythmia-database-1.0.0
print("get data:" + number + " ECG")
record = wfdb.rdrecord('./mit-bih-arrhythmia-database-1.0.0/' + number, channel_names=['MLII'])
record = wfdb.rdrecord('./MIT-BIT/' + number, channel_names=['MLII'])
data = record.p_signal.flatten()
# get R sample and symbol
annotation = wfdb.rdann('mit-bih-arrhythmia-database-1.0.0/' + number, 'atr')
annotation = wfdb.rdann('MIT-BIT/' + number, 'atr')
Rlocatation = annotation.sample
Rclass = annotation.symbol