仅在TensorBoard上获得黑色图像

Question

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

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

---

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

（（1，28，28，1）

img[0][20]

array（[[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

---

有人可以向我解释吗？

Answer 1

实际上，我找到了解决方法

img = img.astype(np.uint8)

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

使用：

.astype(np.uint8)

我可以看到图像