本文主要是介绍小波降噪基础-python版本,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
这篇小文将使用小波多分辨分析对一个简单信号进行降噪,主要是降噪流程,为以后的小波更复杂的降噪算法打下良好的基础。降噪算法流程大致如下:
(1)去趋势项(如直流电流),并将数据归一化到区[0, 1];
(2)进行多级小波分解;
(3)使用步骤 (2)中的细节系数 cD 确定合适的阈值,给出5种不同的方法确定阈值;
(4)将简单的软阈值或硬阈值方法应用于细节系数;
(5)重建信号。
阈值确定方法,更多的细节请查看相关论文,很多
1. universal
在这种情况下,阈值由公式 MAD x sqrt{2 x log(m)} 给出,其中 MAD 是中值绝对偏差,m 是信号的长度。
2. sqtwolog
和universal一样,只是不使用MAD。
3. energy
在这种情况下,阈值算法估计细节系数的能量,并使用它们来估计最佳阈值。
4. stein
此方法实现了 Stein 的无偏风险估计。
5. heurstein
这是 Stein 的无偏风险估计的启发式实现。
首先导入相关模块
import numpy as np
import pandas as pd
import matplotlib.pylab as plt
from scipy.signal import butter, filtfilt
from scipy.signal import spectrogram
from denoising import WaveletDenoising
写个函数用于绘制所有小波分解的系数
def plot_coeffs_distribution(coeffs):fig = plt.figure()size_ = int(len(coeffs) // 2) + 1if size_ % 2 != 0:size_ = size_+1for i in range(len(coeffs)):ax = fig.add_subplot(size_, 2, i+1)ax.hist(coeffs[i], bins=50)def pretty_plot(data, titles, palet, fs=1, length=100, nperseg=256):fig = plt.figure(figsize=(13, 13))fig.subplots_adjust(hspace=0.5, wspace=0.5)index = 1for i, d in enumerate(data):ax = fig.add_subplot(8, 2, index)ax.plot(d[:length], color=palet[i])ax.set_title(titles[i])ax = fig.add_subplot(8, 2, index+1)f, t, Sxx = spectrogram(d, fs=fs, nperseg=nperseg)ax.pcolormesh(t, f, Sxx, shading='auto')index += 2
5种阈值方法的计算函数
def run_experiment(data, level=2, fs=1, nperseg=256, length=100):titles = ['Original data','Universal Method','SURE Method','Energy Method','SQTWOLOG Method','Heursure Method']experiment = ['universal','stein','energy','sqtwolog','heurstein']wd = WaveletDenoising(normalize=False,wavelet='db3',level=level,thr_mode='soft',selected_level=level,method="universal",resolution=100,energy_perc=0.90)res = [data]for i, e in enumerate(experiment):wd.method = experiment[i]res.append(wd.fit(data))palet = ['r', 'b', 'k', 'm', 'c', 'orange', 'g', 'y']pretty_plot(res,titles,palet,fs=fs,length=length,nperseg=nperseg)
搞个最简单的数据,以便于初级分析
if __name__ == '__main__':data = np.zeros((128,))data[:16] = 4data += np.random.normal(0, 1, (128,))run_experiment(data, level=3, length=100, nperseg=32)plt.show()
各阈值的计算没有列出,本文只是讲个小波降噪的大体流程。
完整代码:
小波降噪基础-python版本
工学博士,担任《Mechanical System and Signal Processing》审稿专家,担任《中国电机工程学报》优秀审稿专家,《控制与决策》,《系统工程与电子技术》,《电力系统保护与控制》,《宇航学报》等EI期刊审稿专家。
擅长领域:现代信号处理,机器学习,深度学习,数字孪生,时间序列分析,设备缺陷检测、设备异常检测、设备智能故障诊断与健康管理PHM等。
这篇关于小波降噪基础-python版本的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!