From 12cb80fb45d0f19c5d98ee60cda346ad324d1377 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Wed, 3 Jan 2018 18:01:19 +0900 Subject: true alignment --- become_yukarin/dataset/dataset.py | 6 +++--- 1 file changed, 3 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 b597bba..d259734 100644 --- a/become_yukarin/dataset/dataset.py +++ b/become_yukarin/dataset/dataset.py @@ -237,12 +237,12 @@ class DecodeFeatureProcess(BaseDataProcess): def __call__(self, data: numpy.ndarray, test): data = data.T - lens = [self._sizes[t] for t in self._targets] - assert data.shape[1] == sum(lens) + lasts = numpy.cumsum([self._sizes[t] for t in self._targets]).tolist() + assert data.shape[1] == lasts[-1] d = defaultdict(lambda: numpy.nan, **{ t: data[:, bef:aft] - for t, bef, aft in zip(self._targets, [0] + lens[:-1], lens) + for t, bef, aft in zip(self._targets, [0] + lasts[:-1], lasts) }) return AcousticFeature( f0=d['f0'], -- cgit v1.2.3-70-g09d2