Pytorch torchvision.utils.save_image 图片保存工具

图片保存工具

# 图片张数
N = 16
# 通道数
C = 3
# 高度
H = 64
# 宽度
W = 32
image = torch.rand(N,C,H,W)
# 路径
path = './'
save_image(image, path)