superres

1 files changed, 5 insertions, 2 deletions

diff --git a/become_yukarin/acoustic_converter.py b/become_yukarin/acoustic_converter.py
index 62eacff..706c6ac 100644
--- a/become_yukarin/acoustic_converter.py
+++ b/become_yukarin/acoustic_converter.py
@@ -122,17 +122,20 @@ class AcousticConverter(object):
     def convert_from_audio_path(self, path: Path, out_sampling_rate: Optional[int] = None):
         wave = self._wave_process(str(path), test=True)
         feature = self._feature_process(wave, test=True)
-        return self.convert_from_feature(feature, out_sampling_rate)
+        f0 = feature.f0
+        return self.convert_from_feature(feature, out_sampling_rate, f0)
 
     def convert_from_feature_path(self, path: Path, out_sampling_rate: Optional[int] = None):
         feature = self._acoustic_feature_load_process(path, test=True)
         return self.convert_from_feature(feature, out_sampling_rate)
 
-    def convert_from_feature(self, input: AcousticFeature, out_sampling_rate: Optional[int] = None):
+    def convert_from_feature(self, input: AcousticFeature, out_sampling_rate: Optional[int] = None, f0=None):
         if out_sampling_rate is None:
             out_sampling_rate = self.config.dataset.param.voice_param.sample_rate
 
         out = self.convert_to_feature(input=input, out_sampling_rate=out_sampling_rate)
+        if f0 is not None:
+            out.f0 = f0
         out = pyworld.synthesize(
             f0=out.f0.ravel(),
             spectrogram=out.spectrogram,