如何在 Pytorch 中保存灰度图像？

Question

我想在 Pytorch 中保存灰度图像，每个图像有四个灰度值0，60、120和180。我尝试以下方法来保存图像，但保存的图像不是我所期望的。

for i, (inputs) in enumerate(test_generator):

    pred = modelPl(inputs.float()).detach()

    fig,ax = plt.subplots(1,1,figsize = (5,5))

    ax.imshow(pred[0,:,:], cmap = "gray")

    print(pred.shape)

    torchvision.utils.save_image(pred, saveTestPath + 'img_{0}.png'.format(i)) 

Output: torch.Size([400, 400])

Expected image:

但是，保存的图片不正确如下：

Answer 1

可能torchvision.utils.save_image需要值在 0 到 1 的范围内。您的图像的值大于 1，因此存在问题。您可以通过将张量除以255（或某个适当的数字）来检查这一点。您也可以尝试设置一下normalize=True，看看是否可以自动为您归一化数据。