From cf71104424ca3234da06abf4ce6e5794fda594c5 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Fri, 24 Nov 2017 06:43:24 +0900 Subject: atr subset scirpts --- become_yukarin/dataset/dataset.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'become_yukarin/dataset/dataset.py') diff --git a/become_yukarin/dataset/dataset.py b/become_yukarin/dataset/dataset.py index 329226c..dae2324 100644 --- a/become_yukarin/dataset/dataset.py +++ b/become_yukarin/dataset/dataset.py @@ -126,21 +126,28 @@ class AcousticFeatureLoadProcess(BaseDataProcess): class AcousticFeatureSaveProcess(BaseDataProcess): - def __init__(self, validate=False): + def __init__(self, validate=False, ignore: List[str] = None): self._validate = validate + self._ignore = ignore if ignore is not None else [] def __call__(self, data: Dict[str, any], test=None): path = data['path'] # type: Path feature = data['feature'] # type: AcousticFeature if self._validate: feature.validate() - numpy.save(path.absolute(), dict( + + d = dict( f0=feature.f0, spectrogram=feature.spectrogram, aperiodicity=feature.aperiodicity, mfcc=feature.mfcc, voiced=feature.voiced, - )) + ) + for k in self._ignore: + assert k in d + d[k] = numpy.nan + + numpy.save(path.absolute(), d) class DistillateUsingFeatureProcess(BaseDataProcess): -- cgit v1.2.3-70-g09d2