Pythorch将一小批4D张量保存为图像

2024-03-29 10:54:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我有8张量在4D张量形状(B,C,H,W)的集合。更具体地说,我的张量的形状是(8,3,64,64),也就是说,8张图像的格式是[3x64x64]。在

我想把它们保存在1.png,2.png…,8.png等中。当我尝试torchvision.utils.save_images(my_tensor)时,这是很好的工作。但是,这是将图像保存在单个网格中,而不是单个图像中。在

我如何解决这个问题?在

我试过下面的代码,但没用。在

for i in range(tensor.size(0)):

        np_data = tensor.cpu().numpy()
        imgplot = plt.imshow(np_data)

Tags: 代码图像网格fordatapngmysave
1条回答
网友
1楼 · 发布于 2024-03-29 10:54:12

如果要保存单个图像,可以使用:

for i in range(tensor.size(0)):
           torchvision.utils.save_image(tensor[i, :, :, :], '{}.png'.format(i))

将另存为:1.png,2.png…,8.png

相关问题 更多 >