| | import argparse |
| | import os |
| | from pathlib import Path |
| |
|
| | from synthesizer.hparams import hparams |
| | from synthesizer.synthesize import run_synthesis |
| | from utils.argutils import print_args |
| |
|
| |
|
| |
|
| | if __name__ == "__main__": |
| | class MyFormatter(argparse.ArgumentDefaultsHelpFormatter, argparse.RawDescriptionHelpFormatter): |
| | pass |
| |
|
| | parser = argparse.ArgumentParser( |
| | description="Creates ground-truth aligned (GTA) spectrograms from the vocoder.", |
| | formatter_class=MyFormatter |
| | ) |
| | parser.add_argument("datasets_root", type=Path, help=\ |
| | "Path to the directory containing your SV2TTS directory. If you specify both --in_dir and " |
| | "--out_dir, this argument won't be used.") |
| | parser.add_argument("-s", "--syn_model_fpath", type=Path, |
| | default="saved_models/default/synthesizer.pt", |
| | help="Path to a saved synthesizer") |
| | parser.add_argument("-i", "--in_dir", type=Path, default=argparse.SUPPRESS, help= \ |
| | "Path to the synthesizer directory that contains the mel spectrograms, the wavs and the " |
| | "embeds. Defaults to <datasets_root>/SV2TTS/synthesizer/.") |
| | parser.add_argument("-o", "--out_dir", type=Path, default=argparse.SUPPRESS, help= \ |
| | "Path to the output vocoder directory that will contain the ground truth aligned mel " |
| | "spectrograms. Defaults to <datasets_root>/SV2TTS/vocoder/.") |
| | parser.add_argument("--hparams", default="", help=\ |
| | "Hyperparameter overrides as a comma-separated list of name=value pairs") |
| | parser.add_argument("--cpu", action="store_true", help=\ |
| | "If True, processing is done on CPU, even when a GPU is available.") |
| | args = parser.parse_args() |
| | print_args(args, parser) |
| | modified_hp = hparams.parse(args.hparams) |
| |
|
| | if not hasattr(args, "in_dir"): |
| | args.in_dir = args.datasets_root / "SV2TTS" / "synthesizer" |
| | if not hasattr(args, "out_dir"): |
| | args.out_dir = args.datasets_root / "SV2TTS" / "vocoder" |
| |
|
| | if args.cpu: |
| | |
| | os.environ["CUDA_VISIBLE_DEVICES"] = "-1" |
| |
|
| | run_synthesis(args.in_dir, args.out_dir, args.syn_model_fpath, modified_hp) |
| |
|
