pytorch中torchvision.utils包下的save_image函数保存空白图像问题

代码引用https://www.cnblogs.com/hypnus-ly/p/12922597.html

import torch
import torchvision
import numpy as np
from PIL import Image

# 读取rgb格式的图片
img = Image.open("E:/code/python/medfilter/1lena.png")
# (h,w,c)--(c,h,w)，pytorch的输入格式必须为(c,h,w)维度换位
img = np.array(img).transpose((2,0,1))

# 执行了transpose后，numpy数组的内存不连续了，转换到tensor时会报错，需要先执行如下操作
img = np.ascontiguousarray([img])
img = torch.Tensor(img)

# 以下两句代码可以注释，save_image()函数里已经包含了make_grid()操作
img_grid = torchvision.utils.make_grid(img)
print(img_grid.shape)

# img如果没有归一化，必须要除以255。
# torchvision.utils.save_image(img,"E:/code/python/medfilter/test.png")#图像为空白
torchvision.utils.save_image(img/255.0,"E:/code/python/medfilter/test.png")