【Pytorch】torchvision.utils.save_image直接保存tensor为图片

一般来说,需要将tensor转变为numpy类型的数组从而保存图片,这样的过程比较繁琐,Pytorch提供了save_image()函数,可直接将tensor保存为图片,若tensor在cuda上也会移到CPU中进行保存。 ) 参数: tensor (Tensor or list): Image to be saved. If given a mini-batch tensor, saves th
相关文章
相关标签/搜索