pytorch tensor 保存图片

自带的

from torchvision.utils import save_image
img=im_data.data
img-=img.min()
img/=img.max()
img*=255
img=img.cpu()
img=img.squeeze().
npimg=img.permute(1,2,0).numpy().astype('uint8')
plt.imsave('data/images/'+str(uuid1())+'.jpg',npimg)

你可能感兴趣的:(DL+ML)