mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
mrq 2024-08-05 20:38:29 -05:00
parent 134dac8c2b
commit 8bac8fe902
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/process_libritts.py
View File

@ -109,12 +109,15 @@ def process(
txts = [] txts = []
wavs = [] wavs = []
for book_id in os.listdir(f'./{input_audio}/{dataset_name}/{speaker_id}'): for book_id in os.listdir(f'./{input_audio}/{group_name}/{speaker_id}'):
if not os.path.isdir(f'./{input_audio}/{group_name}/{speaker_id}/{book_id}'): if not os.path.isdir(f'./{input_audio}/{group_name}/{speaker_id}/{book_id}'):
print("Is not dir:", f'./{input_audio}/{group_name}/{speaker_id}/{book_id}') print("Is not dir:", f'./{input_audio}/{group_name}/{speaker_id}/{book_id}')
continue continue
for filename in os.listdir(f'./{input_audio}/{dataset_name}/{speaker_id}/{book_id}'): for filename in os.listdir(f'./{input_audio}/{group_name}/{speaker_id}/{book_id}'):
if ".wav" not in filename:
continue
inpath = Path(f'./{input_audio}/{group_name}/{speaker_id}/{book_id}/{filename}') inpath = Path(f'./{input_audio}/{group_name}/{speaker_id}/{book_id}/{filename}')
if not inpath.exists(): if not inpath.exists():
missing["audio"].append(str(inpath)) missing["audio"].append(str(inpath))