diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/extract_spectrogram_pair.py | 2 | ||||
| -rw-r--r-- | scripts/super_resolution_test.py | 12 |
2 files changed, 11 insertions, 3 deletions
diff --git a/scripts/extract_spectrogram_pair.py b/scripts/extract_spectrogram_pair.py index be21459..f2b96b4 100644 --- a/scripts/extract_spectrogram_pair.py +++ b/scripts/extract_spectrogram_pair.py @@ -29,6 +29,7 @@ parser.add_argument('--pad_second', type=float, default=base_voice_param.pad_sec parser.add_argument('--frame_period', type=int, default=base_acoustic_feature_param.frame_period) parser.add_argument('--order', type=int, default=base_acoustic_feature_param.order) parser.add_argument('--alpha', type=float, default=base_acoustic_feature_param.alpha) +parser.add_argument('--f0_estimating_method', default=base_acoustic_feature_param.f0_estimating_method) parser.add_argument('--enable_overwrite', action='store_true') arguments = parser.parse_args() @@ -53,6 +54,7 @@ def generate_file(path): frame_period=arguments.frame_period, order=arguments.order, alpha=arguments.alpha, + f0_estimating_method=arguments.f0_estimating_method, ) feature = acoustic_feature_process(wave, test=True).astype_only_float(numpy.float32) high_spectrogram = feature.spectrogram diff --git a/scripts/super_resolution_test.py b/scripts/super_resolution_test.py index 8b04ce0..4f34632 100644 --- a/scripts/super_resolution_test.py +++ b/scripts/super_resolution_test.py @@ -18,10 +18,12 @@ parser.add_argument('model_names', nargs='+') parser.add_argument('-md', '--model_directory', type=Path, default=Path('/mnt/dwango/hiroshiba/become-yukarin/')) parser.add_argument('-iwd', '--input_wave_directory', type=Path, default=Path('/mnt/dwango/hiroshiba/become-yukarin/dataset/yukari-wave/yukari-news/')) +parser.add_argument('-g', '--gpu', type=int) args = parser.parse_args() model_directory = args.model_directory # type: Path input_wave_directory = args.input_wave_directory # type: Path +gpu = args.gpu paths_test = list(Path('./test_data_sr/').glob('*.wav')) @@ -41,6 +43,7 @@ def process(p: Path, super_resolution: SuperResolution): frame_period=param.acoustic_feature_param.frame_period, order=param.acoustic_feature_param.order, alpha=param.acoustic_feature_param.alpha, + f0_estimating_method=param.acoustic_feature_param.f0_estimating_method, ) try: @@ -68,7 +71,7 @@ for model_name in args.model_names: model_paths = base_model.glob('predictor*.npz') model_path = list(sorted(model_paths, key=extract_number))[-1] print(model_path) - super_resolution = SuperResolution(config, model_path) + super_resolution = SuperResolution(config, model_path, gpu=gpu) output = Path('./output').absolute() / base_model.name output.mkdir(exist_ok=True) @@ -76,5 +79,8 @@ for model_name in args.model_names: paths = [path_train, path_test] + paths_test process_partial = partial(process, super_resolution=super_resolution) - pool = multiprocessing.Pool() - pool.map(process_partial, paths) + if gpu is None: + pool = multiprocessing.Pool() + pool.map(process_partial, paths) + else: + list(map(process_partial, paths)) |