[WIP] add f0

1 files changed, 13 insertions, 0 deletions

diff --git a/become_yukarin/data_struct.py b/become_yukarin/data_struct.py
index 63043e2..7b220f0 100644
--- a/become_yukarin/data_struct.py
+++ b/become_yukarin/data_struct.py
@@ -1,5 +1,7 @@
 from typing import NamedTuple
 
+import pyworld
+
 import numpy
 
 
@@ -23,3 +25,14 @@ class AcousticFeature(NamedTuple):
             mfcc=self.mfcc.astype(dtype),
             voiced=self.mfcc.astype(dtype),
         )
+
+    @staticmethod
+    def get_sizes(sampling_rate: int, order: int):
+        fft_size = pyworld.get_cheaptrick_fft_size(fs=sampling_rate)
+        return dict(
+            f0=1,
+            spectrogram=fft_size // 2 + 1,
+            aperiodicity=fft_size // 2 + 1,
+            mfcc=order + 1,
+            voiced=1,
+        )