Al Lot

2020-10-07 15:21:45 +02:00
parent 5848b528f0
commit f296ba78b9
6 changed files with 78 additions and 39 deletions
--- a/audio_toolset/mel_augmentation.py
+++ b/audio_toolset/mel_augmentation.py
@ -59,9 +59,9 @@ class ShiftTime(object):
            # Set to silence for heading/ tailing
            shift = int(shift)
            if shift > 0:
-                augmented_data[:, :shift] = 0
+                augmented_data[:shift, :] = 0
            else:
-                augmented_data[:, shift:] = 0
+                augmented_data[shift:, :] = 0
            return augmented_data
        else:
            return x