Tensorboard add_image显示绿色图像红色

Question

我遇到了一个很奇怪的问题。在通过我的神经网络运行了一些图像并尝试显示如下分段后：

print(label.shape)

torch.Size（[1，3，321，321]）

现在使用matplotlib显示图像可以正确显示所有内容：

plt.imshow(fake_seg_tb[0].permute(1,2,0))

将图像放入张量板（pytorch API）时，会得到图像的红色版本，这不是像素值。

writer_semisuper = SummaryWriter()
writer_semisuper.add_image('My Label: ', torchvision.utils.make_grid(label), some_step)

我要导入的库是（PyTorch 1.5.0）：

from torch.utils.tensorboard import SummaryWriter

与tensorboard --version的蜜蜂2.1.0一样。

我不知道为什么在matplotlib和张量中RGB值正确，但是当我显示它时，这是不正确的。