python 灰度图的保存方式_pytorch学习系列(5):单通道灰度图的保存(大坑)

pytorch的torchvision无法保存真正的灰度图像

在图像相关的深度学习模型中,有时候需要保存训练中的图像或者测试得到的图像(尤其是低级视觉任务比如去噪、超分辨率、压缩重建等),一般使用如下方式进行图像保存(torchvision.utils中的save_image()函数):

torchvision.utils.save_image(output.data,'%d.bmp'% (idx), padding=0)

但这种方式只能保存RGB彩色图像,如果网络的输出是单通道灰度图像,则该函数依然会输出三个通道,每个通道的数值都是相同的,即“伪灰度图像”,虽然从视觉效果上看不出区别,但是图像所占内存比正常情况大了两倍。

那么如何保存真正的灰度图像?首先转入save_image()函数进行探究:

def save_image(tensor, filename, nrow=8, padding=2,

normalize=False, range=None, scale_each=False, pad_value=0):

"""Save a given Tensor into an image file.

Args:

tensor (Tensor or list): Image to be saved. If given a mini-batch tensor,

saves the tensor as a grid of images by calling ``make_grid``.

**kwargs: Other arguments are documented in ``make_grid``

  • 1
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值