Skip to content
Snippets Groups Projects
Commit 40c5f1dc authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Use matplotlib to save images

parent d25aaf5f
Branches
Tags
1 merge request!37Use matplotlib to save images
Pipeline #32742 passed
# Code referenced from https://gist.github.com/gyglim/1f8dfb1b5c82627ae3efcfbbadb9f514
import tensorflow as tf
import numpy as np
import scipy.misc
import matplotlib.pyplot as plt
try:
from StringIO import StringIO # Python 2.7
except ImportError:
......@@ -29,7 +29,7 @@ class Logger(object):
s = StringIO()
except:
s = BytesIO()
scipy.misc.toimage(img).save(s, format="png")
plt.imsave(s, img, format='png')
# Create an Image object
img_sum = tf.Summary.Image(encoded_image_string=s.getvalue(),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment