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

dataset stretch now optional

Browse Source
This commit is contained in:
Si11ium
2020-05-15 09:43:29 +02:00
parent 2c4d903fa2
commit d929149a15
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
audio_toolset/audio_augmentation.py
View File

@ -9,6 +9,8 @@ class Speed(object):
self.max_ratio = max_ratio self.max_ratio = max_ratio
def __call__(self, x): def __call__(self, x):
if not all([self.speed_factor, self.max_ratio]):
return x
start = int(np.random.randint(0, x.shape[-1],1)) start = int(np.random.randint(0, x.shape[-1],1))
end = int((np.random.uniform(0, self.max_ratio, 1) * x.shape[-1]) + start) end = int((np.random.uniform(0, self.max_ratio, 1) * x.shape[-1]) + start)
end = min(end, x.shape[-1]) end = min(end, x.shape[-1])