diff options
| author | Jan Buethe <jbuethe@amazon.de> | 2023-11-08 16:43:46 +0300 |
|---|---|---|
| committer | Jan Buethe <jbuethe@amazon.de> | 2023-11-08 16:43:46 +0300 |
| commit | b852f7272c839e96c4d27370507e2269412027db (patch) | |
| tree | ae39247bf29cd5791c6dc394798d4515c36f3335 | |
| parent | ba26645119a2198942f1663a909bb625ea21d924 (diff) | |
removed offset from dataset
| -rw-r--r-- | dnn/torch/osce/data/silk_enhancement_set.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/dnn/torch/osce/data/silk_enhancement_set.py b/dnn/torch/osce/data/silk_enhancement_set.py index 65e97508..fd18c4de 100644 --- a/dnn/torch/osce/data/silk_enhancement_set.py +++ b/dnn/torch/osce/data/silk_enhancement_set.py @@ -49,7 +49,6 @@ class SilkEnhancementSet(Dataset): num_bands_noisy_spec=18, noisy_spec_scale='opus', noisy_apply_dct=True, - add_offset=False, add_double_lag_acorr=False, ): @@ -73,7 +72,6 @@ class SilkEnhancementSet(Dataset): self.gains = np.fromfile(os.path.join(path, 'features_gain.f32'), dtype=np.float32) self.num_bits = np.fromfile(os.path.join(path, 'features_num_bits.s32'), dtype=np.int32) self.num_bits_smooth = np.fromfile(os.path.join(path, 'features_num_bits_smooth.f32'), dtype=np.float32) - self.offsets = np.fromfile(os.path.join(path, 'features_offset.f32'), dtype=np.float32) self.clean_signal_hp = np.fromfile(os.path.join(path, 'clean_hp.s16'), dtype=np.int16) self.clean_signal = np.fromfile(os.path.join(path, 'clean.s16'), dtype=np.int16) @@ -86,7 +84,6 @@ class SilkEnhancementSet(Dataset): num_bands_noisy_spec, noisy_spec_scale, noisy_apply_dct, - add_offset, add_double_lag_acorr) self.history_len = 700 if add_double_lag_acorr else 350 @@ -120,8 +117,7 @@ class SilkEnhancementSet(Dataset): self.lpcs[frame_start : frame_stop], self.gains[frame_start : frame_stop], self.ltps[frame_start : frame_stop], - self.periods[frame_start : frame_stop], - self.offsets[frame_start : frame_stop] + self.periods[frame_start : frame_stop] ) if self.preemph > 0: |