mirror of
https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git
synced 2026-10-18 11:23:39 +08:00
load two layer of cnn
This commit is contained in:
1 parent
815bd40d40
commit
5f757ad939
11 files changed
+158
-126
No files matched your search
@@ -0,0 +1,31 @@
|
||||
import numpy as np
|
||||
import pywt
|
||||
import wfdb
|
||||
from matplotlib import pyplot as plt
|
||||
|
||||
# file error
|
||||
|
||||
# use db5 ,level9 do deviose
|
||||
ECGDATAPATH = './MIT-BIT/'
|
||||
|
||||
# from ECG_read_weiyuqing_without_wfdb import PATH
|
||||
# record = wfdb.rdrecord(ECGDATAPATH + number, channel_names=['MLII'])
|
||||
record = wfdb.rdrecord(ECGDATAPATH + '100', channel_names=['MLII'])
|
||||
data = record.p_signal.flatten()
|
||||
|
||||
coeffs = pywt.wavedec(data=data, wavelet='db5', level=9)
|
||||
cA9, cD9, cD8, cD7, cD6, cD5, cD4, cD3, cD2, cD1 = coeffs
|
||||
|
||||
threshold = (np.median(np.abs(cD1)) / 0.6745) * (np.sqrt(2 * np.log(len(cD1))))
|
||||
cD1.fill(0)
|
||||
cD2.fill(0)
|
||||
for i in range(1, len(coeffs) - 2):
|
||||
coeffs[i] = pywt.threshold(coeffs[i], threshold)
|
||||
|
||||
rdata = pywt.waverec(coeffs=coeffs, wavelet='db5')
|
||||
|
||||
plt.figure(figsize=(20, 4))
|
||||
plt.plot(data)
|
||||
plt.show()
|
||||
plt.plot(rdata)
|
||||
plt.show()
|
||||
Reference in new issue
Block a user