Python的EEMD实现

PyEMD包的安装

注意，不能直接pip install PyEMD
官方：/en/latest/
正确安装：pip install EMD-signal

EEMD代码

# -*- coding: UTF-8 -*-

import librosa
import numpy as np
from pyemd import EEMD
# 导包PyEMD会报错，所以将所有的PyEMD都替换成pyemd，包括安装的包里面的
import pylab as plt


def signal_eemd(S, RATE):
    T = 0.05
    t = np.linspace(0,T, num=int(RATE * T))

    eemd = EEMD()
    eemd.trials = 500
    eemd.noise_seed(12345)
	# 这里的S和t跟采样率和原始音频时长有关，RATE * T为音频的信号长度
    eIMFs = eemd.eemd(S, t, -1)
    nIMFs = eIMFs.shape[0]
    plt.figure(figsize=(12, 9))
    plt.subplot(nIMFs + 1, 1, 1)
    plt.plot(t, S, 'r')

    for n in range(nIMFs):
        plt.subplot(nIMFs + 1, 1, n + 2)
        plt.plot(t, eIMFs[n], 'g')
        plt.ylabel("eIMF %i" % (n + 1))
        plt.locator_params(axis='y', nbins=5)

    plt.xlabel("Time [s]")
    plt.tight_layout()
    plt.savefig('eemd_example', dpi=120)
    plt.show()
    plt.plot(t, eIMFs[0] + eIMFs[1] + eIMFs[1], 'g')
    plt.show()

if __name__ == "__main__":
	RATE = 80000	# 采样率
    X, sr = librosa.load('', sr=RATE)
    signal_eemd(X, RATE)