From 0b410c0bbacc147950438423835547b74859aac1 Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Fri, 2 Feb 2018 16:59:33 +0900 Subject: pix2pix convertモデル MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- become_yukarin/acoustic_converter.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'become_yukarin/acoustic_converter.py') diff --git a/become_yukarin/acoustic_converter.py b/become_yukarin/acoustic_converter.py index 498bdb1..62eacff 100644 --- a/become_yukarin/acoustic_converter.py +++ b/become_yukarin/acoustic_converter.py @@ -73,6 +73,9 @@ class AcousticConverter(object): input = self._feature_normalize(input, test=True) input = self._encode_feature(input, test=True) + pad = 128 - input.shape[1] % 128 + input = numpy.pad(input, [(0, 0), (0, pad)], mode='minimum') + converter = partial(chainer.dataset.convert.concat_examples, device=self.gpu, padding=0) inputs = converter([input]) @@ -81,6 +84,7 @@ class AcousticConverter(object): if self.gpu is not None: out = chainer.cuda.to_cpu(out) + out = out[:, :-pad] out = self._decode_feature(out, test=True) out = AcousticFeature( -- cgit v1.2.3-70-g09d2