def test_large_chain(self):
chain = audaugio.FlatChain(audaugio.PitchShiftAugmentation(-1),
audaugio.TimeStretchAugmentation(.95),
audaugio.EqualizerAugmentation(300, 1, 2),
audaugio.BackgroundNoiseAugmentation(.005),
audaugio.TimeStretchAugmentation(1.25))
augmented = chain(self.signal, self.sr)
self.assertEqual(len(augmented), 5)
def get_augmentation_chains():
windowing_augmentation = audaugio.WindowingAugmentation(4, 2)
imitation_augmentations = audaugio.CombinatoricChain(
audaugio.BackgroundNoiseAugmentation(.005),
audaugio.LowPassAugmentation(500, 1.5, 1),
audaugio.HighPassAugmentation(6000, 1.5, 1), windowing_augmentation)
reference_augmentations = audaugio.LinearChain(windowing_augmentation)
return imitation_augmentations, reference_augmentations
def setUp(self):
super().setUp()
self.amplitude = .005
self.augmentor = audaugio.BackgroundNoiseAugmentation(self.amplitude)
import librosa
import audaugio
y, sr = librosa.load('sample.wav')
chain = audaugio.CombinatoricChain(
audaugio.PitchShiftAugmentation(1),
audaugio.BackgroundNoiseAugmentation(.005),
audaugio.EqualizerAugmentation(800, .15, -15))
augmented_audio = chain(y, sr)
for i, a in enumerate(augmented_audio):
librosa.output.write_wav("output/{0}.wav".format(i), a, sr)