From 41d64ffb02aae0a61cb41044757b2dd6b7b783c1 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 23 Jul 2018 16:16:58 +0200 Subject: process --- scripts/process.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'scripts/process.py') diff --git a/scripts/process.py b/scripts/process.py index 8f5c2f0..3a2b788 100644 --- a/scripts/process.py +++ b/scripts/process.py @@ -10,7 +10,8 @@ import numpy from become_yukarin import AcousticConverter from become_yukarin import SuperResolution -from become_yukarin.config.config import create_from_json as create_config +from become_yukarin.config.config import create_from_json as create_conv_config +from become_yukarin.config.sr_config import create_from_json as create_sr_config parser = argparse.ArgumentParser() parser.add_argument('-cmd', '--conv_model_directory', type=Path, default=Path('/mnt/dwango/hiroshiba/become-yukarin/')) @@ -33,7 +34,7 @@ def extract_number(f): return int(s[-1]) if s else -1 def load_acoustic_converter(): - conv_config = create_config(conv_model / 'config.json') + conv_config = create_conv_config(conv_model / 'config.json') if it is not None: conv_model_path = conv_model / 'predictor_{}.npz'.format(it) else: @@ -43,7 +44,7 @@ def load_acoustic_converter(): return AcousticConverter(conv_config, conv_model_path, gpu=gpu) def load_super_resolution(): - sr_config = create_config(sr_model / 'config.json') + sr_config = create_sr_config(sr_model / 'config.json') sr_model_paths = sr_model.glob('predictor*.npz') sr_model_path = list(sorted(sr_model_paths, key=extract_number))[-1] print(sr_model_path) -- cgit v1.2.3-70-g09d2