def test_contrast(self): """ Test contrast effect, compare to SoX implementation """ enhancement_amount = 80. 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("contrast", [enhancement_amount]) sox_output_waveform, sr = E.sox_build_flow_effects() waveform, sample_rate = torchaudio.load(noise_filepath, normalization=True) output_waveform = F.contrast(waveform, enhancement_amount) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def test_contrast(self): """ Test contrast effect, compare to SoX implementation """ enhancement_amount = 80. E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(self.noise_filepath) E.append_effect_to_chain("contrast", [enhancement_amount]) sox_output_waveform, sr = E.sox_build_flow_effects() output_waveform = F.contrast(self.noise_waveform, enhancement_amount) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def func(tensor): enhancement_amount = 80. return F.contrast(tensor, enhancement_amount)
def test_contrast(self): enhancement_amount = 80. data, path = self.get_whitenoise() result = F.contrast(data, enhancement_amount) self.assert_sox_effect(result, path, ['contrast', enhancement_amount])