def test_overdrive(self): gain = 30 colour = 40 data, path = self.get_whitenoise() result = F.overdrive(data, gain, colour) self.assert_sox_effect(result, path, ['overdrive', gain, colour])
def test_overdrive(self): """ Test overdrive effect, compare to SoX implementation """ gain = 30 colour = 40 noise_filepath = common_utils.get_asset_path('whitenoise.wav') E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(noise_filepath) E.append_effect_to_chain("overdrive", [gain, colour]) sox_output_waveform, sr = E.sox_build_flow_effects() waveform, _ = torchaudio.load(noise_filepath, normalization=True) output_waveform = F.overdrive(waveform, gain, colour) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def test_overdrive(self): """ Test overdrive effect, compare to SoX implementation """ gain = 30 colour = 40 E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(self.noise_filepath) E.append_effect_to_chain("overdrive", [gain, colour]) sox_output_waveform, sr = E.sox_build_flow_effects() output_waveform = F.overdrive(self.noise_waveform, gain, colour) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def func(tensor): gain = 30. colour = 50. return F.overdrive(tensor, gain, colour)