PySDKit: signal decomposition in Python

License

A Python library for signal decomposition algorithms 🥳

Installation | Example Script | Target | Acknowledgements

Installation 🚀 <a id="Installation"></a>

You can install PySDKit through pip:

pip install pysdkit

We only used NumPy, Scipy and matplotlib when developing the project.

Example Script ✨ <a id="Example-Script"></a>

This project integrates simple signal processing methods, signal decomposition and visualization, and builds a general interface similar to Scikit-learn. It is mainly divided into three steps:

Import the signal decomposition method;
Create an instance for signal decomposition;
Use the fit_transform method to implement signal decomposition;
Visualize and analyze the original signal and the intrinsic mode functions IMFs obtained by decomposition.

from pysdkit import EMD
from pysdkit.data import test_emd
from pysdkit.plot import plot_IMFs

t, signal = test_emd()

# create an instance for signal decomposition
emd = EMD()
# implement signal decomposition
IMFs = emd.fit_transform(signal, max_imfs=2)
plot_IMFs(signal, IMFs)

example

The EMD in the above example is the most classic empirical mode decomposition algorithm in signal decomposition. For more complex signals, you can try other algorithms such as variational mode decomposition (VMD).

import numpy as np
from pysdkit import VMD

# load new signal
signal = np.load("./example/example.npy")

# use variational mode decomposition
vmd = VMD(alpha=500, K=3, tau=0.0, tol=1e-9)
IMFs = vmd.fit_transform(signal=signal)
print(IMFs.shape)

vmd.plot_IMFs(save_figure=True)

vmd_example

Better observe the characteristics of the decomposed intrinsic mode function in the frequency domain.

from pysdkit.plot import plot_IMFs_amplitude_spectra

# frequency domain visualization
plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth

frequency_example

Target 🎯 <a id="Target"></a>

PySDKit is still under development. We are currently working on reproducing the signal decomposition algorithms in the table below, including not only common decomposition algorithms for univariate signals such as EMD and VMD, but also decomposition algorithms for multivariate signals such as MEMD and MVMD. We will also further reproduce the decomposition algorithms for two-dimensional images to make PySDKit not only suitable for signal processing, but also for image analysis and understanding. See Mission for the reasons why we developed PySDKit.

PySDKit

Install / Use

README

PySDKit: signal decomposition in Python

Installation 🚀 <a id="Installation"></a>

Example Script ✨ <a id="Example-Script"></a>

Target 🎯 <a id="Target"></a>