如何在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，看看它是否可以自动为您标准化数据。