Tensorboard：尝试记录 RGB 图像。类型错误：无法处理此数据类型

Question

我正在尝试将 RGB 图像记录到 Tensorboard。我确保我的图像是一个形状为 (206,127,3) 的 numpy 数组。里面的值是numpy.uint8，范围是0-255。

当我这样做时：

writer.add_image("Example Image", img, epoch)

我收到错误：

TypeError: Cannot handle this data type: (1, 1, 206), |u1

该错误似乎来自tensorboard内部使用的PIL模块。不过，我可以使用 PIL 手动保存此图像。

Answer 1

我刚刚在tensorboarX文档中找到了答案： https://tensorboardx.readthedocs.io/en/latest/tutorial.html#add-image

预计形状为 (3, H, W)，而不是我的形状 (H, W, 3)。