added dithering to input signal

xiph · Sep 15, 2024 · fa589bc · fa589bc
1 parent 0f49d5d
commit fa589bc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dnn/torch/osce/data/simple_bwe_dataset.py b/dnn/torch/osce/data/simple_bwe_dataset.py
@@ -74,6 +74,10 @@ def __getitem__(self, index):
  x_16 = self.signal_16k[signal_start16 : signal_stop16].astype(np.float32) / 2**15
  history_16 = self.signal_16k[signal_start16 - 320 : signal_start16].astype(np.float32) / 2**15
 
+ # dithering
+ x_16 += (np.random.rand(len(x_16)) - 0.5) / 2**15
+ history_16 += (np.random.rand(len(history_16)) - 0.5) / 2**15
+
  x_48 = self.signal_48k[3 * signal_start16 - self.upsampling_delay48
  : 3 * signal_stop16 - self.upsampling_delay48].astype(np.float32) / 2**15