def compute_and_check_summary_pb(self, name, images, max_outputs=3, images_tensor=None, feed_dict=None): """Use both `op` and `pb` to get a summary, asserting equality. Returns: a `Summary` protocol buffer """ if images_tensor is None: images_tensor = tf.cast(tf.constant(images), tf.uint8) op = summary.op(name, images_tensor, max_outputs=max_outputs) pb = summary.pb(name, images, max_outputs=max_outputs) pb_via_op = self.pb_via_op(op, feed_dict=feed_dict) self.assertProtoEquals(pb, pb_via_op) return pb
def image(self, *args, **kwargs): return summary.pb(*args, **kwargs)
def test_requires_rank_4_in_pb(self): with six.assertRaisesRegex(self, ValueError, 'must have rank 4'): summary.pb('mona_lisa', np.array([[1, 2, 3], [4, 5, 6]]))