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