Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
3 years ago
import torch
import torchvision
from torchvision.datasets import MNIST
import matplotlib.pyplot as plt
%matplotlib inline
dataset= MNIST(root='data/', download=True)
Downloading http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz Downloading http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz to data/MNIST\raw\train-images-idx3-ubyte.gz 
  0%|          | 0/9912422 [00:00<?, ?it/s]
Extracting data/MNIST\raw\train-images-idx3-ubyte.gz to data/MNIST\raw Downloading http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz Downloading http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz to data/MNIST\raw\train-labels-idx1-ubyte.gz 
  0%|          | 0/28881 [00:00<?, ?it/s]
Extracting data/MNIST\raw\train-labels-idx1-ubyte.gz to data/MNIST\raw Downloading http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz Downloading http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz to data/MNIST\raw\t10k-images-idx3-ubyte.gz 
  0%|          | 0/1648877 [00:00<?, ?it/s]
Extracting data/MNIST\raw\t10k-images-idx3-ubyte.gz to data/MNIST\raw Downloading http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz Downloading http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz to data/MNIST\raw\t10k-labels-idx1-ubyte.gz 
  0%|          | 0/4542 [00:00<?, ?it/s]
Extracting data/MNIST\raw\t10k-labels-idx1-ubyte.gz to data/MNIST\raw
C:\Users\akifr\anaconda3\lib\site-packages\torchvision\datasets\mnist.py:498: UserWarning: The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors. This means you can write to the underlying (supposedly non-writeable) NumPy array using the tensor. You may want to copy the array to protect its data or make it writeable before converting it to a tensor. This type of warning will be suppressed for the rest of this program. (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:180.) return torch.from_numpy(parsed.astype(m[2], copy=False)).view(*s) 
test_dataset= MNIST(root='data/', train=False)
dataset[0]
(<PIL.Image.Image image mode=L size=28x28 at 0x18D41AC2610>, 5)
image, label = dataset[0]
plt.imshow(image, cmap='gray')
print('Label:', label)
Label: 5
Notebook Image