Тренируя свои сети, я хочу визуализировать видео каждой эпохи.До сих пор я делал функцию, которая получает мой тензор (размер: [3, 32, 64,64] - размер 3 пакета, 32 кадра, разрешение 64X64).Я хочу визуализировать некоторые видео в виде сетки.Я пытался использовать
import torch
import torchvision.utils as vutils
import numpy as np
import matplotlib.pyplot as plt
import torchvision.transforms as transforms
import scipy.misc
def show(img):
npimg = img.numpy()
plt.imshow(np.transpose(npimg, (1,2,0))interpolation='nearest')
video = torch.randn(3,32,64,64)
grid = vutils.make_grid(video)
show(grid) # here to use save_img func
Но это не удалось.кто нибудь знает как это сделать?Спасибо!