mirror of
https://github.com/babysor/MockingBird.git
synced 2024-03-22 13:11:31 +08:00
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
from pathlib import Path
|
|
from toolbox import Toolbox
|
|
from utils.argutils import print_args
|
|
from utils.modelutils import check_model_paths
|
|
import argparse
|
|
import os
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description="Runs the toolbox",
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
|
)
|
|
|
|
parser.add_argument("-d", "--datasets_root", type=Path, help= \
|
|
"Path to the directory containing your datasets. See toolbox/__init__.py for a list of "
|
|
"supported datasets.", default=None)
|
|
parser.add_argument("-e", "--enc_models_dir", type=Path, default="encoder/saved_models",
|
|
help="Directory containing saved encoder models")
|
|
parser.add_argument("-s", "--syn_models_dir", type=Path, default="synthesizer/saved_models",
|
|
help="Directory containing saved synthesizer models")
|
|
parser.add_argument("-v", "--voc_models_dir", type=Path, default="vocoder/saved_models",
|
|
help="Directory containing saved vocoder models")
|
|
parser.add_argument("--cpu", action="store_true", help=\
|
|
"If True, processing is done on CPU, even when a GPU is available.")
|
|
parser.add_argument("--seed", type=int, default=None, help=\
|
|
"Optional random number seed value to make toolbox deterministic.")
|
|
parser.add_argument("--no_mp3_support", action="store_true", help=\
|
|
"If True, no mp3 files are allowed.")
|
|
args = parser.parse_args()
|
|
print_args(args, parser)
|
|
|
|
if args.cpu:
|
|
# Hide GPUs from Pytorch to force CPU processing
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
|
del args.cpu
|
|
|
|
## Remind the user to download pretrained models if needed
|
|
check_model_paths(encoder_path=args.enc_models_dir, synthesizer_path=args.syn_models_dir,
|
|
vocoder_path=args.voc_models_dir)
|
|
|
|
# Launch the toolbox
|
|
Toolbox(**vars(args))
|