Python Tensorboard SummaryWriter add_image（）函数中的反转颜色

有一个图像存储在

image\u tensor

（image\u tensor的大小

（3256512）

，存储间隔

0255

）中，我希望通过SummaryWriter的add\u image（）函数将其显示在Tensorboard（更具体地说，对于PyTorch是TensorboardX）中。当我通过

writer.add\u image（“color\u image”，image\u tensor，self.step）将图像添加到张力板中时，

，颜色会反转

当我通过

scipy.misc.imsave（“/write/to/path/image.png”、np.transpose（image\u tensor.data.cpu（）.numpy（），（1,2,0）））

将图像写入文件时，图像非常好

---

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

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

将图像插入matplotlib图形，然后使用

add\u 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中工作