From 4732160ed647e99efdafb14f60b36e98c0dc67a3 Mon Sep 17 00:00:00 2001 From: iver56 Date: Mon, 17 Aug 2020 15:38:56 +0200 Subject: [PATCH] Move dB conversion in Gain from apply to randomize_parameters --- audiomentations/augmentations/transforms.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/audiomentations/augmentations/transforms.py b/audiomentations/augmentations/transforms.py index 04a5ab4d..0180074d 100644 --- a/audiomentations/augmentations/transforms.py +++ b/audiomentations/augmentations/transforms.py @@ -775,12 +775,9 @@ def __init__(self, min_gain_in_db=-12, max_gain_in_db=12, p=0.5): def randomize_parameters(self, samples, sample_rate): super().randomize_parameters(samples, sample_rate) if self.parameters["should_apply"]: - self.parameters["gain_in_db"] = random.uniform( - self.min_gain_in_db, self.max_gain_in_db + self.parameters["amplitude_ratio"] = convert_decibels_to_amplitude_ratio( + random.uniform(self.min_gain_in_db, self.max_gain_in_db) ) def apply(self, samples, sample_rate): - amplitude_ratio = convert_decibels_to_amplitude_ratio( - self.parameters["gain_in_db"] - ) - return samples * amplitude_ratio + return samples * self.parameters["amplitude_ratio"]