python小波变换去噪-ECG信号

原文请见:https://blog.csdn.net/weixin_41322458/article/details/91385984

一,小波去噪原理:
信号产生的小波系数含有信号的重要信息,将信号经小波分解后小波系数较大,噪声的小波系数较小,并且噪声的小波系数要小于信号的小波系数,通过选取一个合适的阀值,大于阀值的小波系数被认为是有信号产生的,应予以保留,小于阀值的则认为是噪声产生的,置为零从而达到去噪的目的。

小波阀值去噪的基本问题包括三个方面:小波基的选择,阀值的选择,阀值函数的选择。
(1) 小波基的选择:通常我们希望所选取的小波满足以下条件:正交性、高消失矩、紧支性、对称性或反对称性。但事实上具有上述性质的小波是不可能存在的,因为小波是对称或反对称的只有Haar小波,并且高消失矩与紧支性是一对矛盾,所以在应用的时候一般选取具有紧支的小波以及根据信号的特征来选取较为合适的小波。
(2) 阀值的选择:直接影响去噪效果的一个重要因素就是阀值的选取,不同的阀值选取将有不同的去噪效果。目前主要有通用阀值(VisuShrink)、SureShrink阀值、Minimax阀值、BayesShrink阀值等。
(3) 阀值函数的选择:阀值函数是修正小波系数的规则,不同的反之函数体现了不同的处理小波系数的策略。最常用的阀值函数有两种:一种是硬阀值函数,另一种是软阀值函数。还有一种介于软、硬阀值函数之间的Garrote函数。

另外,对于去噪效果好坏的评价,常用信号的信噪比(SNR)与估计信号同原始信号的均方根误差(RMSE)来判断。

import matplotlib.pyplot as plt
import pywt
import numpy as np
import os

#load data.txt
print("------Load ecg.txt------")
#加载数据为ndarray
ecg=np.loadtxt(‘C:\Users\lenovo\Desktop\lv fei hua 0100120191218.txt’,delimiter=’ ',dtype=int)
ecg = list(ecg)
ecg = ecg[1:1000]
#print(ecg)
print(len(ecg))

#ecg = pywt.data.ecg() # 生成心电信号

index = []
data = []
for i in range(len(ecg)-1):
X = float(i)
Y = float(ecg[i])
index.append(X)
data.append(Y)

Create wavelet object and define parameters

w = pywt.Wavelet(‘db8’) # 选用Daubechies8小波
maxlev = pywt.dwt_max_level(len(data), w.dec_len)
print("maximum level is " + str(maxlev))
threshold = 0.04 # Threshold for filtering

Decompose into wavelet components, to the level selected:

coeffs = pywt.wavedec(data, ‘db8’, level=maxlev) # 将信号进行小波分解

plt.figure()
for i in range(1, len(coeffs)):
coeffs[i] = pywt.threshold(coeffs[i], threshold*max(coeffs[i])) # 将噪声滤波

datarec = pywt.waverec(coeffs, ‘db8’) # 将信号进行小波重构

mintime = 0
maxtime = mintime + len(data) + 1

plt.figure()
plt.subplot(2, 1, 1)
plt.plot(index[mintime:maxtime], data[mintime:maxtime])
plt.xlabel(‘time (s)’)
plt.ylabel(‘microvolts (uV)’)
plt.title(“Raw signal”)
plt.subplot(2, 1, 2)
plt.plot(index[mintime:maxtime], datarec[mintime:maxtime-1])
plt.xlabel(‘time (s)’)
plt.ylabel(‘microvolts (uV)’)
plt.title(“De-noised signal using wavelet techniques”)

plt.tight_layout()
plt.show()
在这里插入图片描述
下一步是,做去噪效果好坏的评价,常用信号的信噪比(SNR)与估计信号同原始信号的均方根误差(RMSE)来判断。

发布了34 篇原创文章 · 获赞 4 · 访问量 1766

猜你喜欢

转载自blog.csdn.net/qq_36495569/article/details/104086636
今日推荐