diff --git a/logger.py b/logger.py index 3dee3b1..b529530 100644 --- a/logger.py +++ b/logger.py @@ -1,5 +1,7 @@ from io import BytesIO +import PIL +from PIL import Image import scipy.misc import tensorflow as tf @@ -16,7 +18,7 @@ def scalar_summary(self, tag, value, step): def image_summary(self, tag, image, step): s = BytesIO() - scipy.misc.toimage(image).save(s, format="png") + Image.fromarray(image).save(s, format="png") # Create an Image object img_sum = tf.Summary.Image( @@ -36,7 +38,7 @@ def image_list_summary(self, tag, images, step): img_summaries = [] for i, img in enumerate(images): s = BytesIO() - scipy.misc.toimage(img).save(s, format="png") + Image.fromarray(img).save(s, format="png") # Create an Image object img_sum = tf.Summary.Image( diff --git a/requirements.txt b/requirements.txt index 8d18acd..c8accbe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,6 +5,6 @@ scikit-image==0.14.2 imageio==2.5.0 medpy==0.4.0 Pillow==6.0.0 -scipy==1.2.1 +scipy==1.3.0 pandas==0.24.2 -tqdm==4.32.1 \ No newline at end of file +tqdm==4.32.1