Fix args.out_file

wkentaro · Jan 7, 2018 · a7454ce · a7454ce
1 parent 08af1cc
commit a7454ce
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/pytorch2chainer/infer_pytorch.py b/examples/pytorch2chainer/infer_pytorch.py
@@ -138,6 +138,5 @@ def get_z(mu, logvar):
     viz.append(fake_B)
 viz = fcn.utils.get_tile_image(viz)
 
-out_file = osp.join(here, 'logs/infer_pytorch.png')
-cv2.imwrite(out_file, viz[:, :, ::-1])
-print('Saved file: %s' % out_file)
+cv2.imwrite(args.out_file, viz[:, :, ::-1])
+print('Saved file: %s' % args.out_file)