diff options
Diffstat (limited to 'scripts/process.py')
| -rw-r--r-- | scripts/process.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/process.py b/scripts/process.py index 47f013f..b169f26 100644 --- a/scripts/process.py +++ b/scripts/process.py @@ -24,6 +24,7 @@ parser.add_argument('-iwd', '--input_directory', type=Path, default=Path('./inpu parser.add_argument('-owd', '--output_directory', type=Path, default=Path('./output')) parser.add_argument('-it', '--iteration', type=int) parser.add_argument('-g', '--gpu', type=int) +parser.add_argument('-f0', '--convert_f0', action='store_true') args = parser.parse_args() conv_model = args.conv_model_directory # type: Path @@ -32,6 +33,7 @@ input_directory = args.input_directory output_directory = args.output_directory it = args.iteration gpu = args.gpu +convert_f0 = args.convert_f0 conv_config = create_conv_config(conv_model / 'config.json') sr_config = create_sr_config(sr_model / 'config.json') @@ -73,7 +75,7 @@ def process(p: Path, acoustic_converter: AcousticConverter, super_resolution: Su # converted_feature.f0 = input_feature.f0.astype(numpy.float64) merged_feature = AcousticFeature( - f0=input_feature.f0, + f0=converted_feature.f0 if convert_f0 else input_feature.f0, spectrogram=converted_feature.spectrogram, aperiodicity=converted_feature.aperiodicity, mfcc=converted_feature.mfcc, |