torch_ecg.augmenters

Augmenters are used to augment the training data.

AugmenterManager(*augs[, random])

The Module to manage the augmenters.

Augmenter()

Base class for augmenters.

BaselineWanderAugmenter(fs[, bw_fs, ...])

Generate baseline wander composed of sinusoidal and Gaussian noise.

CutMix([fs, num_mix, alpha, beta, prob, inplace])

CutMix augmentation.

LabelSmooth([fs, smoothing, prob, inplace])

Label smoothing augmentation.

Mixup([fs, alpha, beta, prob, inplace])

Mixup augmentor.

RandomFlip([fs, per_channel, prob, inplace])

Randomly flip the ECGs along the voltage axis.

RandomMasking(fs[, mask_value, mask_width, ...])

Randomly mask ECGs with a probability.

RandomRenormalize([mean, std, per_channel, ...])

Randomly re-normalize the ECG tensor, using the Z-score normalization method.

StretchCompress([ratio, prob, inplace])

Stretch-or-compress augmenter on ECG tensors.

StretchCompressOffline([ratio, prob, ...])

Offline stretch-or-compress augmenter.