Use imsave from skimage instead of scipy.

Fixes #177.
inferno-pytorch · Jun 6, 2019 · 564c9cd · 564c9cd
1 parent 5158e7d
commit 564c9cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inferno/utils/io_utils.py b/inferno/utils/io_utils.py
@@ -2,7 +2,7 @@
 import h5py as h5
 import numpy as np
 import yaml
-from scipy.misc import imsave
+from skimage.io import imsave
 
 
 # Function to load in a dataset from a h5file
@@ -61,7 +61,7 @@ def _print_image(image, prefix, batch, channel, z=None):
         else:
             file_name = "{}--B-{}--CH-{}--Z-{}.png".format(prefix, batch, channel, z)
         full_file_name = os.path.join(directory, file_name)
-        imsave(arr=image, name=full_file_name)
+        imsave(arr=image, fname=full_file_name)
 
     for batch in range(tensor.shape[0]):
         for channel in range(tensor.shape[1]):