From f126040da2df8c343a2f63219008808cd596f81e Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 5 Aug 2021 23:24:38 -0600 Subject: [PATCH] Undo noise first --- codes/data/audio/wav_aug.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/data/audio/wav_aug.py b/codes/data/audio/wav_aug.py index 9c8a870d..a35646d0 100644 --- a/codes/data/audio/wav_aug.py +++ b/codes/data/audio/wav_aug.py @@ -40,14 +40,14 @@ class WavAugmentor: ] band_effect = random.choice(band_effects) volume_effects = [ - ['loudness', rdi(10,-10)], + ['loudness', rdi(10,-2)], ['overdrive', rdi(20,0), rdi(20,0)], ] vol_effect = random.choice(volume_effects) effects = [speed_effect, band_effect, vol_effect] + out, sr = torchaudio.sox_effects.apply_effects_tensor(wav, sample_rate, effects) # Add a variable amount of noise - out = wav + torch.rand_like(wav) * random.random() * .05 - out, sr = torchaudio.sox_effects.apply_effects_tensor(out, sample_rate, effects) + out = out + torch.rand_like(out) * random.random() * .05 return out