Tensorboard SummaryWriter add_image() 函数中的反转颜色

Question

有一个图像存储在image_tensor（大小为(3,256,512) 的图像张量，在区间0,255 中存储值）我想通过 add_image() 在 Tensorboard（PyTorch 的 TensorboardX，更具体地说）中显示SummaryWriter 的函数。当我通过writer.add_image("color_image",image_tensor,self.step) 将图像添加到 Tensorboard 时，颜色会反转。

当我通过scipy.misc.imsave("/write/to/path/image.png",np.transpose(image_tensor.data.cpu().numpy(),(1,2,0)))将图像写入文件时，图像非常好。

我对第二行的唯一更改是将 CxHxW 更改为 HxWxC，但我认为这不是颜色反转问题的根源。可能是什么问题？

Answer 1

我遇到了类似的问题，我也无法跟踪问题。对我有用但有点麻烦的解决方案是：

获取您的图像并将其插入 matplotlib 图形，然后使用 add_figure。

例如：

fig, ax =  plt.subplots(2,3)
# add your subplots with some images eg.
ax[0,0].imshow(image_1)
# etc.
writer_semisuper.add_figure("testfig", fig, 0)

这显示了与您创建的完全相同的图，但分辨率较低。因此，如果您的绘图在 jupyter 中工作或保存在磁盘上，它也应该在 tensorboard 中工作。