Permalink
Browse files

Add requirements.txt file

  • Loading branch information...
jilljenn committed Dec 10, 2018
2 parents 475014e + 7498b29 commit f4d2e60864586da3f67a8e7d70b17713ff3dcf32
Showing with 6 additions and 1 deletion.
  1. +1 −1 i2v/chainer_i2v.py
  2. +5 −0 requirements.txt
@@ -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

@@ -0,0 +1,5 @@
numpy
scipy
Pillow
scikit-image
chainer>=2.0.0

0 comments on commit f4d2e60

Please sign in to comment.