def WriteImageSeries(writer, tag, n_images=1): """Write a few dummy images to writer.""" step = 0 session = session_lib.Session() p = array_ops.placeholder("uint8", (1, 4, 4, 3)) s = logging_ops.image_summary(tag, p) for _ in xrange(n_images): im = np.random.random_integers(0, 255, (1, 4, 4, 3)) summ = session.run(s, feed_dict={p: im}) writer.add_summary(summ, step) step += 20 session.close()
def testImageSummary(self): with self.cached_session(): i = array_ops.ones((5, 4, 4, 3)) s = logging_ops.image_summary('tag', i) self.assertEqual(s.op.type, u'ImageSummary')