steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

Browse Source
This commit is contained in:
steffen
2020-05-15 19:48:07 +02:00
parent c494366691 5a940e8a6a
commit 33df856595
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
audio_toolset/mel_augmentation.py
View File

@@ -13,8 +13,8 @@ class NoiseInjection(object):
def __call__(self, x: np.ndarray): def __call__(self, x: np.ndarray):
if self.noise_factor: if self.noise_factor:
noise = np.random.normal(loc=self.mu, scale=self.sigma, size=x.shape) noise = np.random.uniform(0, self.noise_factor, size=x.shape)
augmented_data = x + self.noise_factor * noise augmented_data = x + x * noise
# Cast back to same data type # Cast back to same data type
augmented_data = augmented_data.astype(x.dtype) augmented_data = augmented_data.astype(x.dtype)
return augmented_data return augmented_data

1
utils/config.py
View File

@@ -39,6 +39,7 @@ class Config(ConfigParser, ABC):
h = hashlib.md5() h = hashlib.md5()
params = deepcopy(self.as_dict) params = deepcopy(self.as_dict)
del params['model']['type'] del params['model']['type']
del params['model']['secondary_type']
del params['data']['worker'] del params['data']['worker']
del params['main'] del params['main']
h.update(str(params).encode()) h.update(str(params).encode())