Commit 40c5f1dc authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Use matplotlib to save images

parent d25aaf5f
Pipeline #32742 passed with stage
in 56 minutes and 1 second
# Code referenced from
import tensorflow as tf
import numpy as np
import scipy.misc
import matplotlib.pyplot as plt
from StringIO import StringIO # Python 2.7
except ImportError:
......@@ -29,7 +29,7 @@ class Logger(object):
s = StringIO()
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(),
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment