Skip to content
This repository was archived by the owner on Apr 19, 2023. It is now read-only.

Commit 564c9cd

Browse files
Use imsave from skimage instead of scipy.
Fixes #177.
1 parent 5158e7d commit 564c9cd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

inferno/utils/io_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import h5py as h5
33
import numpy as np
44
import yaml
5-
from scipy.misc import imsave
5+
from skimage.io import imsave
66

77

88
# Function to load in a dataset from a h5file
@@ -61,7 +61,7 @@ def _print_image(image, prefix, batch, channel, z=None):
6161
else:
6262
file_name = "{}--B-{}--CH-{}--Z-{}.png".format(prefix, batch, channel, z)
6363
full_file_name = os.path.join(directory, file_name)
64-
imsave(arr=image, name=full_file_name)
64+
imsave(arr=image, fname=full_file_name)
6565

6666
for batch in range(tensor.shape[0]):
6767
for channel in range(tensor.shape[1]):

0 commit comments

Comments
 (0)