From 701f1d6220ffef291ac82c1cfacfbde40371a5cc Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Tue, 23 Oct 2018 19:34:07 +0900 Subject: [PATCH 1/2] to chainerV2 --- i2v/chainer_i2v.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/i2v/chainer_i2v.py b/i2v/chainer_i2v.py index 30ee815f..dcd99233 100644 --- a/i2v/chainer_i2v.py +++ b/i2v/chainer_i2v.py @@ -4,9 +4,10 @@ import numpy as np from scipy.ndimage import zoom from skimage.transform import resize +import chainer from chainer import Variable from chainer.functions import average_pooling_2d, sigmoid -from chainer.functions.caffe import CaffeFunction +from chainer.links.caffe import CaffeFunction class ChainerI2V(Illustration2VecBase): @@ -47,7 +48,9 @@ def _forward(self, inputs, layername): input_ -= self.mean # subtract mean input_ = input_.transpose((0, 3, 1, 2)) # (N, H, W, C) -> (N, C, H, W) x = Variable(input_) - y, = self.net(inputs={'data': x}, outputs=[layername], train=False) + with chainer.using_config('train', False): + with chainer.using_config('enable_backprop', False): + y, = self.net(inputs={'data': x}, outputs=[layername]) return y def _extract(self, inputs, layername): From bf08d7fa17adfb274c12b44343a5e57581f84ebb Mon Sep 17 00:00:00 2001 From: Hiroshiba Kazuyuki Date: Wed, 24 Oct 2018 20:42:53 +0900 Subject: [PATCH 2/2] stack with --- i2v/chainer_i2v.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/i2v/chainer_i2v.py b/i2v/chainer_i2v.py index dcd99233..c5d5dcd3 100644 --- a/i2v/chainer_i2v.py +++ b/i2v/chainer_i2v.py @@ -48,9 +48,8 @@ def _forward(self, inputs, layername): input_ -= self.mean # subtract mean input_ = input_.transpose((0, 3, 1, 2)) # (N, H, W, C) -> (N, C, H, W) x = Variable(input_) - with chainer.using_config('train', False): - with chainer.using_config('enable_backprop', False): - y, = self.net(inputs={'data': x}, outputs=[layername]) + with chainer.using_config('train', False), chainer.using_config('enable_backprop', False): + y, = self.net(inputs={'data': x}, outputs=[layername]) return y def _extract(self, inputs, layername):