/ecg-augmentation

Augmentation scheme for the alteration of ECG signals for use in machine learning

Primary LanguagePythonMIT LicenseMIT

ECG Augmentation

WIP

ECG Augmentation is a Python library for augmenting 12-lead ECG signals.

Installation

Use the package manager pip to install ECG Augmentation.

pip install ecgaugmentation

Usage

import ecgaugmentation

in_path = "." # path to -fecg.npy ECG files
iterations = 100
anomaly_type = "RBBB"

ecgs, annotations = ecgaugmentation.augment(in_path, anomaly_type, iterations)

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

License

MIT