mirror of https://github.com/coqui-ai/TTS.git
34 lines
1022 B
Python
34 lines
1022 B
Python
import os
|
|
import json
|
|
|
|
from datasets.preprocess import get_preprocessor_by_name
|
|
|
|
|
|
def make_speakers_json_path(out_path):
|
|
"""Returns conventional speakers.json location."""
|
|
return os.path.join(out_path, "speakers.json")
|
|
|
|
|
|
def load_speaker_mapping(out_path):
|
|
"""Loads speaker mapping if already present."""
|
|
try:
|
|
with open(make_speakers_json_path(out_path)) as f:
|
|
return json.load(f)
|
|
except FileNotFoundError:
|
|
return {}
|
|
|
|
|
|
def save_speaker_mapping(out_path, speaker_mapping):
|
|
"""Saves speaker mapping if not yet present."""
|
|
speakers_json_path = make_speakers_json_path(out_path)
|
|
with open(speakers_json_path, "w") as f:
|
|
json.dump(speaker_mapping, f, indent=4)
|
|
|
|
|
|
def get_speakers(data_root, meta_file, dataset_type):
|
|
"""Returns a sorted, unique list of speakers in a given dataset."""
|
|
preprocessor = get_preprocessor_by_name(dataset_type)
|
|
items = preprocessor(data_root, meta_file)
|
|
speakers = {e[2] for e in items}
|
|
return sorted(speakers)
|