نمایش لیستی از تنسورها در pytorch

Question

سلام . من به صورت زیر دیتاست MNIST را لود کردم.حالا میخام یک batch ار تصاویر را نمایش بدم، ببیتم درست لود شده یا نه یا کلا داده ها به چه صورت هستن 

import torch
import torchvision
from torchvision import datasets, transforms

opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
urllib.request.install_opener(opener)

device_ = torch.device("cuda" if torch.cuda.is_available() else "cpu")

train_data =datasets.MNIST(root='.', train=True, download=True,
				   transform=transforms.Compose([
					   transforms.ToTensor(),
					   transforms.Normalize((0.1307,), (0.3081,))
				   ]))



train_loader_ = torch.utils.data.DataLoader(train_data
	, batch_size=64, shuffle=True, num_workers=4)
dataiter = iter(train_loader_)

 

Answer 1

torchvision.utils متدی به نام make_grid داره که یک Batch از تصاویر رو میگیره و تصاویر را در قالب گرید به شما میده 

که می تونید نمایش بدید.

 

def imshow(img):

    img = img / 2 + 0.5     # unnormalize
    npimg = img.numpy()
    cv2.imshow("view", np.transpose(npimg, (1, 2, 0)))
    cv2.waitKey(0)
	
images, _ = dataiter.next()
imshow(torchvision.utils.make_grid(images))