prepare_align.py

import os
import argparse
import yaml

from preprocessor import libritts, vctk


def main(config):
    if "LibriTTS" in config["dataset"]:
        corpus_path = config["path"]["corpus_path"]
        raw_path = config["path"]["raw_path"]
        dsets = []
        for dmode, dset in config["subsets"].items():
            if dset == "train-clean":
                dsets += ["train-clean-100", "train-clean-360"]
            elif dset == "train-all":
                dsets += ["train-clean-100", "train-clean-360", "train-other-500"]
            # if isinstance(dset, list):
                # dsets += dset
            # elif isinstance(dset, str):
                # dsets.append(dset)
        for dset in dsets:
            config["path"]["corpus_path"] = os.path.join(corpus_path, dset)
            config["path"]["raw_path"] = os.path.join(raw_path, dset)
            libritts.prepare_align(config)
    if "VCTK" in config["dataset"]:
        corpus_path = config["path"]["corpus_path"]
        raw_path = config["path"]["raw_path"]
        for dmode, dset in config["subsets"].items():
            # config["path"]["corpus_path"] = corpus_path
            config["path"]["raw_path"] = os.path.join(raw_path, dset)
            vctk.prepare_align(config)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("config", type=str, help="path to preprocess.yaml")
    args = parser.parse_args()

    config = yaml.load(open(args.config, "r"), Loader=yaml.FullLoader)
    main(config)