pytorch将4D张量的小批量保存为图像

Question

我有一组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)

Answer 1

如果要保存单个图像，可以使用：

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

将另存为：1.png，2.png ...，8.png