将 pytorch 张量保存为 32 位灰度图像

Posted 2023-02-18

【中文标题】将 pytorch 张量保存为 32 位灰度图像

【英文标题】：Saving a pytoch tensor as a 32-bit grayscale Image

【发布时间】：2021-06-29 21:34:21

【问题描述】：

我处理了一个 32 位灰度 .tif 图像，我使用 PIL 将其转换为张量。在此之后，我将其保存为：

torchvision.utils.save_image(train_img_poac,fp=str(j)+".tif")

此方法自动将张量转换为 RGB 格式的图像。我希望我的输出图像是 32 位灰度图像。

我尝试使用 save_image 函数中的参数，但找不到任何东西。是否将其转换为 numpy ndarray，然后将其转换为 32 位图像？

【问题讨论】：

试试tifffile也许...pypi.org/project/tifffile

【参考方案1】：

很遗憾，save_image 没有保留单通道图像的选项。你可以像 OpenCV 一样使用different library：

import cv2

image = train_img_poac.numpy()
cv2.imwrite('image_name', image)