파이토치 MNIST 데이터 셋 불러오기(pytorch mnist)

[파이썬 파이토치] 파이토치 MNIST 데이터 셋 불러오기 : pytorch, torchvision¶

In [1]:

import torch
import matplotlib.pyplot as plt
from torchvision import datasets, transforms

%matplotlib inline

In [2]:

# datasetsMNIST를 이용하여 바로 mnist 데이터 셋을 가져올 수 있습니다 !! 
mnist_dataset = datasets.MNIST('dataset', train=True, download=True, 
               transform=transforms.Compose([
                   transforms.ToTensor()

]))
mnist_dataset

Out[2]:

Dataset MNIST
    Number of datapoints: 60000
    Root location: dataset
    Split: Train
    StandardTransform
Transform: Compose(
               ToTensor()
           )

In [3]:

# 데이터 로더로 전환하여 학습 테스트를 바로 진행할 수 있습니다.
train_loader = torch.utils.data.DataLoader(
    mnist_dataset
)

In [4]:

# 첫번째 이미지만 가져오기
img , label = next(iter(train_loader))

In [5]:

img.shape

Out[5]:

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

In [6]:

# 배치사이즈와 색상 차원 제거
img_show = img[0, 0, :, :]
img_show.shape

Out[6]:

torch.Size([28, 28])

In [7]:

# 샘플 이미지 보여주기
plt.imshow(img_show, 'gray')
plt.show()