如何解决将 pytoch 张量保存为 32 位灰度图像
我处理了一个 32 位灰度 .tif
图像,我使用 PIL 将其转换为张量。在此之后,我将其保存为:
torchvision.utils.save_image(train_img_poac,fp=str(j)+".tif")
此方法自动将张量转换为 RGB 格式的图像。我希望我的输出图像是 32 位灰度图像。
我尝试使用 save_image
函数中的参数,但找不到任何内容。将其转换为 numpy ndarray,然后将其转换为 32 位图像是一种选择吗?
解决方法
遗憾的是,save_image
没有保留单通道图像的选项。您可以使用 different library 之类的 OpenCV:
import cv2
image = train_img_poac.numpy()
cv2.imwrite('image_name',image)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。