python - 使用 PIL 显示图像会导致奇怪的色块

Question

我正在研究输出图像的 pytorch 模型。问题在于显示此图像，因为输出是一个值在 0-1 之间的火炬张量。如果我这样显示它可以正常工作

out=model.forward(x)
out=torch.squeeze(out)
out=out.permute(1,2,0)
plt.imshow(out)

编辑：这是这个输出的样子，这不是实际的输出，因为我现在失去了重量，但对于这个问题的所有目的都是一样的

但是，如果我在显示之前将其转换为 PIL 图像，则会产生如图所示的奇怪色块

out=model.forward(x)
out=torch.squeeze(out)
out=transforms.ToPILImage()(out)
plt.imshow(out)

我没有使用第一种方法，因为在此之后我还有其他一些步骤，这要求我拥有一个整数值在 0-255 之间的图像。

标签： pythonimagecolorspytorchpython-imaging-library