from local

This commit is contained in:
dohe0342 2023-03-13 14:29:09 +09:00
parent ae76dcd4f7
commit 122ab29aab
2 changed files with 3 additions and 5 deletions

View File

@ -59,9 +59,9 @@ def prepare_vox(
# ) # )
manifests = {} manifests = {}
spk_idx = spk_id.split('_')[1]
#dataset_parts = ["train", "dev", "test"] #dataset_parts = ["train", "dev", "test"]
dataset_parts = [spk_idx] dataset_parts = [spk_id]
if output_dir is not None: if output_dir is not None:
output_dir = Path(output_dir) output_dir = Path(output_dir)
output_dir.mkdir(parents=True, exist_ok=True) output_dir.mkdir(parents=True, exist_ok=True)
@ -81,9 +81,7 @@ def prepare_vox(
#part_file_names = list(map(lambda x: x.strip('.wav'), os.listdir(part_path))) #part_file_names = list(map(lambda x: x.strip('.wav'), os.listdir(part_path)))
part_file_names = sorted(glob.glob(str(part_path)+'/*.wav')) part_file_names = sorted(glob.glob(str(part_path)+'/*.wav'))
part_file_names = [name.split('/')[-1].replace('.wav', '') for name in part_file_names] part_file_names = [name.split('/')[-1].replace('.wav', '') for name in part_file_names]
spk_idx = spk_id.split('_')[1] txt_path = os.path.join(corpus_dir, f"../{spk_id}/texts")
txt_path = os.path.join(corpus_dir, f"../{spk_idx}_texts")
print(txt_path)
futures = [] futures = []
for trans_path in tqdm( for trans_path in tqdm(