首页 > 解决方案 > TensorBoard 上只有黑色图像

问题描述

我尝试了 image_summaries.ipynb ,它运行良好并正确显示图像。

然后我写了一些代码,但是图像不显示,我只有一个黑色的图像。


img = np.reshape(train_images[0], (-1, 28, 28, 1))
print(img.shape)

(1, 28, 28, 1)

img[0][20]

数组([[0],[0],[0],[0],[0],[0],[0],[15],[220],[253],[253],[80] ,[0],[0],[0],[0],[0],[0],[0],[0],[0],[0],[0],[0],[ 0],[0],[0],[0]])

!rm -rf logs
logdir = "logs/train_data/" + datetime.now().strftime("%Y%m%d-%H%M%S")
file_writer = tf.summary.create_file_writer(logdir)
with file_writer.as_default():
  tf.summary.image("Training data", img, step=0)
%tensorboard --logdir logs/train_data

有人可以向我解释吗?

黑色图像

标签: tensorboard

解决方案


其实我找到了解决办法

img = img.astype(np.uint8)

即使 [253] 看起来像一个整数,它也可能不是一个整数......

和 :

.astype(np.uint8)

我可以看到图片


推荐阅读