mirror of
https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git
synced 2026-10-17 19:05:30 +08:00
Merge branch 'main' of https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test
This commit is contained in:
commit
d5a44029c7
1 file changed
+2
-2
@@ -5,11 +5,11 @@ import numpy as np
|
|||||||
def getDataset(number, X_data):
|
def getDataset(number, X_data):
|
||||||
# get ECG from mit-bih-arrhythmia-database-1.0.0
|
# get ECG from mit-bih-arrhythmia-database-1.0.0
|
||||||
print("get data:" + number + " ECG")
|
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()
|
data = record.p_signal.flatten()
|
||||||
|
|
||||||
# get R sample and symbol
|
# 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
|
Rlocatation = annotation.sample
|
||||||
Rclass = annotation.symbol
|
Rclass = annotation.symbol
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user