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

【问题标题】：How to save grayscale image in Pytorch?如何在 Pytorch 中保存灰度图像？
【发布时间】：2019-03-27 09:37:07
【问题描述】：

我想在 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:

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

【问题讨论】：

标签： python image computer-vision pytorch

【解决方案1】：

torchvision.utils.save_image 可能要求值在 0 到 1 的范围内。您的图像的值大于 1，因此存在问题。

您可以通过将张量除以255（或某个适当的数字）来检查这一点。你也可以尝试设置normalize=True，看看能不能自动为你规范化数据。

【讨论】：

谢谢，我通过将张量除以 255 来解决问题。