Add requirements.txt file

rezoo · Dec 10, 2018 · f4d2e60 · f4d2e60
2 parents 475014e + 7498b29
commit f4d2e60
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/i2v/chainer_i2v.py b/i2v/chainer_i2v.py
@@ -48,7 +48,7 @@ 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('train', False), chainer.using_config('enable_backprop', False):
             y, = self.net(inputs={'data': x}, outputs=[layername])
         return y
 

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+numpy
+scipy
+Pillow
+scikit-image
+chainer>=2.0.0