Fix for gaps augmentation

mozilla · Apr 24, 2020 · 5e63baf · 5e63baf
1 parent 5123cd2
commit 5e63baf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/training/deepspeech_training/util/signal_augmentations.py b/training/deepspeech_training/util/signal_augmentations.py
@@ -141,7 +141,8 @@ def apply(self, sample, clock):
         n_gaps = pick_value_from_range(self.n_gaps, clock=clock)
         for _ in range(n_gaps):
             size = pick_value_from_range(self.size, clock=clock)
-            offset = max(0, random.randint(0, len(audio) - size - 1))
+            size = min(size, len(audio) // 10)  # a gap should never exceed 10 percent of the audio
+            offset = random.randint(0, max(0, len(audio) - size - 1))
             audio[offset:offset + size] = 0
         sample.audio = audio