From cee44f61c90c1e7cb543ed3e8148b905c3c0e554 Mon Sep 17 00:00:00 2001
From: jules on spawn <spawn@cortex>
Date: Sun, 29 Jul 2018 18:10:55 +0200
Subject: process

---
 scripts/process.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'scripts')

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,
-- 
cgit v1.2.3-70-g09d2