write_png¶
- torchvision.io.write_png(input: Tensor, filename: str, compression_level: int = 6)[源代码]¶
接收一个 CHW 布局（灰度图像为 HW 布局）的输入张量，并将其保存为 PNG 文件。
- 参数:
input (Tensor[通道数, 图像高度, 图像宽度]) –
c通道的 int8 图像张量，其中c必须为 1 或 3。filename (str 或
pathlib.Path) – 保存图像的路径。compression_level (int) – 生成文件的压缩因子，必须是 0 到 9 之间的数字。默认为 6
