summary_writer.py 文件源码

def image_summary(self, tag, image):
        image = image.reshape((1, image.shape[0], image.shape[1], 1)).astype(np.float32)

        image_summary_op = tf.image_summary(tag, image)
        image_summary_str = tf.Session().run(image_summary_op)

        SummaryWriter().writer.add_summary(image_summary_str, 0)
        SummaryWriter().writer.flush()

        rospy.loginfo("?? " + tag + " image plotted.")
        pass