将tensor转为图像保存

最新推荐文章于 2025-02-18 15:58:05 发布

木琦

最新推荐文章于 2025-02-18 15:58:05 发布

阅读量2.5k

点赞数 2

分类专栏： learn 文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_38335768/article/details/122346278

版权

learn 专栏收录该内容

29 篇文章

订阅专栏

本文对比了两种处理图像残留并将其保存为图片的方法：一种使用PyTorch的可视化工具，另一种借助OpenCV进行numpy操作。讨论了从GPU转储到CPU、保存为PNG和转换为JPG的流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from torchvision import utils as vutils
print('begin')
for j in range(t):                               
    print('begin')
    from torchvision import utils as vutils
    residue=residues[:,i,:,:,:]
    pic = residue.clone().detach()
    pic = pic.to(torch.device('cpu'))
    
    dir='/data/dataset1/wj/RealBasicVSR/results/clean/'             
    filename=dir+str(j)+'.png'                    
    vutils.save_image(pic, filename)
print('over')

第二种方法

print('begin')
import cv2  

for j in range(t):                                           
    residue=residues[:,j,:,:,:]
    data = residue.transpose(1,3)
    
    data = data.cpu().numpy()
    data = data[0,:,:,:]*255		#没有*255之前是黑色的
    print(data.shape)
    name = str(j)+".jpg"
    cv2.imwrite(name,data)