mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

introduce prepadlength

Browse Source
This commit is contained in:
James Betker 2022-05-23 08:21:27 -06:00
parent 10f4a742bd
commit 9f16b25ce5
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/audio/unsupervised_audio_dataset.py
View File

@ -147,6 +147,7 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
# To increase variability, which skew is applied to the clip and resampled_clip is randomized. # To increase variability, which skew is applied to the clip and resampled_clip is randomized.
random.shuffle(skew) random.shuffle(skew)
clips = [] clips = []
prepad_length = min(audio_norm.shape[-1], self.pad_to)
for sk in skew: for sk in skew:
if self.pad_to is not None: if self.pad_to is not None:
if audio_norm.shape[-1] <= self.pad_to: if audio_norm.shape[-1] <= self.pad_to:
@ -159,6 +160,7 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
clips.append(audio_norm) clips.append(audio_norm)
output = { output = {
'prepad_length': prepad_length,
'clip': clips[0], 'clip': clips[0],
'clip_lengths': torch.tensor(audio_norm.shape[-1]), 'clip_lengths': torch.tensor(audio_norm.shape[-1]),
'path': filename, 'path': filename,