diff --git a/augment.py b/augment.py new file mode 100644 index 0000000..3c6fdd2 --- /dev/null +++ b/augment.py @@ -0,0 +1,14 @@ +from audiomentations import Compose, AddGaussianNoise, PitchShift, HighPassFilter +import librosa +import soundfile as sf + +aug = Compose([ + AddGaussianNoise(min_amplitude=0.001, max_amplitude=0.01, p=1.0), + PitchShift(min_semitones=0, max_semitones=0, p=1), + HighPassFilter(min_cutoff_freq=20, max_cutoff_freq=170, p=1) +]) + +def augmentSignal(signal, sr): + aug_sig = aug(signal, sr) + return aug_sig +