Как получить доступ к отдельным точкам данных MNIST и проверить их размер, форму и c из объекта набора поездов с помощью набора данных импорта torchvision

Question

следующий код:

    from torchvision import datasets, transforms
    trainset = datasets.MNIST('./data/', download=True, train=True, transform=transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,))]))

Я хотел бы визуализировать первую точку данных в переменной trainset выше. Я хочу взглянуть на значения пикселей первой точки данных, выполнив что-то вроде print(trainset[0]), или проверить размер, выполнив print(trainset[0].size), или проверить форму, выполнив print(trainset[0].shape) et c.

Answer 1

Чтобы визуализировать данные, вы можете их построить.

import matplotlib.pyplot as plt
plt.imshow(trainset.data[0], cmap='gray')

график 1-го изображения в наборе данных

Чтобы посмотреть значения пикселей 1-го изображения:

print(trainset.data[0])

Чтобы найти форму 1-го изображения:

trainset.data[0].shape
>>>torch.Size([28, 28])

Вместо 0, вы можете заменить его любым i, где i = размер набора данных

Answer 2

Для формы:

trainset.data.shape

torch.Size([60000, 28, 28])

Для первого примера:

trainset.data[0]

tensor([[[0, 0, 0,  ..., 0, 0, 0],
         [0, 0, 0,  ..., 0, 0, 0],
         [0, 0, 0,  ..., 0, 0, 0],
         ...,
         [0, 0, 0,  ..., 0, 0, 0],
         [0, 0, 0,  ..., 0, 0, 0],
         [0, 0, 0,  ..., 0, 0, 0]]], dtype=torch.uint8)