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

问题描述

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

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

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

我尝试使用 save_image 函数中的参数,但找不到任何内容。将其转换为 numpy ndarray,然后将其转换为 32 位图像是一种选择吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)