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