使用Tensorboard显示图片

创建日期:2024-06-21
更新日期:2025-01-31

首先,下载一张png格式的图片(注意:只支持png格式),命名为1.png。然后,打开PythonShell,输入以下代码:

import tensorflow as tf

# 获取图片数据
file = open('1.png', 'rb')
data = file.read()
file.close()

# 图片处理
image = tf.image.decode_png(data, channels=4)
image = tf.expand_dims(image, 0)

# 添加到日志中
sess = tf.Session()
writer = tf.summary.FileWriter('logs')
summary_op = tf.summary.image("image1", image)

# 运行并写入日志
summary = sess.run(summary_op)
writer.add_summary(summary)

# 关闭
writer.close()
sess.close()

然后,在相同目录打开cmd,输入tensorboard ~-~-logdir=logs,然后打开浏览器输入http:~/~/localhost:6006/。在Tensorboard的Images标签页,就可以看到我们的png图片了。

724270-20170211223153963-848108424.png

参考资料:《Tensorflow: How to Display Custom Images in Tensorboard (e.g. Matplotlib Plots)》:http://stackoverflow.com/questions/38543850/tensorflow-how-to-display-custom-images-in-tensorboard-e-g-matplotlib-plots