eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixes

Browse Source
This commit is contained in:
James Betker 2021-09-14 18:29:17 -06:00
parent 4334a67924
commit 9a9c90660f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,7 +16,7 @@ from utils.util import opt_get
def load_audio(audiopath, sampling_rate): def load_audio(audiopath, sampling_rate):
if audiopath[:-4] == '.wav': if audiopath[-4:] == '.wav':
audio, lsr = load_wav_to_torch(audiopath) audio, lsr = load_wav_to_torch(audiopath)
else: else:
audio, lsr = open_audio(audiopath) audio, lsr = open_audio(audiopath)
@ -83,6 +83,9 @@ class UnsupervisedAudioDataset(torch.utils.data.Dataset):
related_files = find_files_of_type('img', os.path.dirname(audiopath), qualifier=is_audio_file)[0] related_files = find_files_of_type('img', os.path.dirname(audiopath), qualifier=is_audio_file)[0]
assert audiopath in related_files assert audiopath in related_files
assert len(related_files) < 50000 # Sanity check to ensure we aren't loading "related files" that aren't actually related. assert len(related_files) < 50000 # Sanity check to ensure we aren't loading "related files" that aren't actually related.
if len(related_files) == 0:
j = 0
print(f"No related files for {audiopath}")
related_files.remove(audiopath) related_files.remove(audiopath)
related_clips = [] related_clips = []
random.shuffle(related_clips) random.shuffle(related_clips)