使用torchvision将张量 保存 为 灰度图

在Python中，使用torchvision中的utils.save_image函数用于保存张量数据为png图片
保存灰度图像与保存3通道PNG图像略有不同。对于灰度图像，你需要注意以下两点：

1. 灰度图像通常是单通道图像，因此在保存时需要确保通道数正确。
2. 灰度图像的每个像素值通常是介于0到255之间的整数。

以下是保存灰度图像的示例代码

import torch
from torchvision import utils

# 创建一个灰度图像张量（单通道）
gray_image = torch.randn(1, 1, 128, 128)  # 假设为128x128的灰度图像

# 将张量值缩放到0到255的范围（整数）
gray_image = ((gray_image + 1) / 2 * 255).clamp(0, 255).byte()

# 使用utils.save_image保存灰度图像
utils.save_image(gray_image, 'gray_image.png')

若张量在GPU上，需要先进行处理放到cpu上，因为最终要保存到内存中

gray_image.data.cpu()

在这个示例中，我们首先创建了一个假设为128x128的灰度图像张量 gray_image，其形状为 [1, 1, 128, 128]。然后，我们通过将张量的值缩放到0到255的范围（整数）并限制在0到255之间，将其转换为合适的灰度图像格式。

最后，我们使用utils.save_image函数将灰度图像保存为PNG文件，命名为gray_image.png。这将会在当前工作目录下保存该文件。

确保在运行代码之前已经安装了torchvision库，该库提供了utils.save_image函数来方便地保存图像。