Python 将pytoch张量保存为32位灰度图像
我已经处理了一个32位灰度Python 将pytoch张量保存为32位灰度图像,python,python-3.x,image-processing,pytorch,save-image,Python,Python 3.x,Image Processing,Pytorch,Save Image,我已经处理了一个32位灰度.tif图像,我使用PIL将其转换为张量。在此之后,我将其保存为: torchvision.utils.save_image(train_img_poac,fp=str(j)+".tif") 此方法会自动将张量转换为RGB格式的图像。我希望我的输出图像是32位灰度图像 我试图使用save_image函数中的参数,但找不到任何内容。将其转换为numpy ndarray,然后将其转换为32位图像是一个选项吗?不幸的是save_Image没有保存单通道图
.tif
图像,我使用PIL将其转换为张量。在此之后,我将其保存为:
torchvision.utils.save_image(train_img_poac,fp=str(j)+".tif")
此方法会自动将张量转换为RGB格式的图像。我希望我的输出图像是32位灰度图像
我试图使用
save_image
函数中的参数,但找不到任何内容。将其转换为numpy ndarray,然后将其转换为32位图像是一个选项吗?不幸的是save_Image
没有保存单通道图像的选项。您可以使用类似OpenCV的:
导入cv2
image=train\u img\u poac.numpy()
cv2.imwrite('image\u name',image)
试试tiffile
也许。。。