Pytorch中四维Tensor转图片并保存（维度顺序调整）

我不是女神ヾ 2022-11-22 10:11 407阅读 0赞

###### 最近在复现一篇论文代码的过程中，想要输出中间图片的结果图，通过debug发现在pytorch网络中是用Tensor存储的四维张量。![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzgxMjYwOQ_size_16_color_FFFFFF_t_70_pic_center] ######

##### 1、维度顺序转换 #####

第一维代表的是batch\_size，然后是通道数和图像尺寸，首先要进行维度顺序的转换  
通过permute函数实现

outputRs = outputR.permute(0,2,3,1)

shape转为96 \* 128 \* 3![在这里插入图片描述][20201103213249426.png_pic_center]

##### 2、转为numpy数组 #####

#由于代码中的中间结果是带有梯度的要进行detach()操作
    		k = outputRs.cpu().detach().numpy()

##### 3、根据第一维度batch\_size逐个读取中间结果，并存储到磁盘中 #####

Image需导入from PIL import Image

for i in range(10):
    			res = k[i] #得到batch中其中一步的图片
    			image = Image.fromarray(np.uint8(res)).convert('RGB')
    			#image.show()
    			#通过时间命名存储结果
    			timestamp = datetime.datetime.now().strftime("%M-%S")
    			savepath = timestamp + '_r.jpg'
    			image.save(savepath)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzgxMjYwOQ_size_16_color_FFFFFF_t_70_pic_center]: /images/20221120/31f2fc864d414f678da6bbd92cde7444.png
[20201103213249426.png_pic_center]: /images/20221120/e1d685df59334c2ba94e3e8c2e1656f9.png

发表评论取消回复

表情：

评论列表（有 0 条评论，407人围观）

还没有评论，来说两句吧...

相关阅读

相关 PyTorch：tensor-张量维度操作（拼接、维度扩展、压缩、转置、重复……）

张量维度操作（拼接、维度扩展、压缩、转置、重复……） note: torch.fun(tensor1)和tensor1.fun()都只会返回改变后的tensor，但是tens

小灰灰/ 2022年12月13日 04:22/ 0 赞/ 712 阅读

相关 Pytorch中四维Tensor转图片并保存（维度顺序调整）

最近在复现一篇论文代码的过程中，想要输出中间图片的结果图，通过debug发现在pytorch网络中是用Tensor存储的四维张量。![在这里插入图片描述][watermark

我不是女神ヾ/ 2022年11月22日 10:11/ 0 赞/ 408 阅读

相关【Pytorch】Tensor维度的拼接与拆分

文章目录 1. 拼接与拆分常用API 2. 按照维度合并Tensor 2.1 cat函数 2.2 stack函数

墨蓝/ 2022年11月03日 04:09/ 0 赞/ 858 阅读

相关【Pytorch】Tensor维度变换

文章目录 Tensor维度变换 1. view / reshape 1.1 view 函数 1.2 resha

我就是我/ 2022年11月02日 13:27/ 0 赞/ 174 阅读

相关 PyTorch教程（四）：维度变换

view和reshape 在PyTorch0.3版本中使用view，在PyTorch0.4以后增加了reshape，作用是将一个shape转变为另一个shape。 >

矫情吗；*/ 2022年10月30日 10:19/ 0 赞/ 216 阅读

相关【PyTorch】 tensor.unsqueeze扩展一个维度

【PyTorch】 tensor.unsqueeze扩展一个维度 1、背景 2、unsqueeze方法 3、示例 1、背景在tensor合并的

小鱼儿/ 2022年08月28日 14:58/ 0 赞/ 247 阅读

相关【PyTorch】 tensor.squeeze压缩维度

【PyTorch】 tensor.squeeze压缩维度 1、背景 2、squeeze方法 3、示例 1、背景 pytorch的tensor维度

爱被打了一巴掌/ 2022年08月28日 14:53/ 0 赞/ 240 阅读

相关 pytorch 获取tensor维度信息

参考[https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor

桃扇骨/ 2022年05月10日 14:38/ 0 赞/ 232 阅读

相关 Pytorch Tensor维度变换

1. 改变shape torch.reshape()、torch.view()可以调整Tensor的shape，返回一个新shape的Tensor，torch.view()是

红太狼/ 2022年01月07日 05:55/ 0 赞/ 289 阅读

相关 pytorch 调整tensor的维度位置

target.permute([0, 3, 1, 2]) 一定要使用permute以及中括号一些在我这里没起到作用的网上的例子： 1. [https

野性酷女/ 2021年12月17日 11:25/ 0 赞/ 319 阅读