Built-in datasets (torchvision.datasets) in PyTorch

Introduction

Built-in datasets help you quickly get real data to practice machine learning without searching or downloading manually.

You want to learn image classification with common datasets like MNIST or CIFAR-10.

You need a quick dataset to test your model code.

You want to compare your model with others using standard datasets.

You want to avoid spending time on data cleaning and focus on model building.

Syntax

PyTorch

from torchvision import datasets

# Load MNIST dataset
mnist_train = datasets.MNIST(root='./data', train=True, download=True, transform=None)

# Load CIFAR-10 dataset
cifar10_train = datasets.CIFAR10(root='./data', train=True, download=True, transform=None)

root is the folder where data will be saved or loaded from.

train=True loads training data; train=False loads test data.

Examples

Load the MNIST test dataset.

PyTorch

from torchvision import datasets

mnist_test = datasets.MNIST(root='./data', train=False, download=True)

Load CIFAR-10 training data and convert images to tensors for PyTorch.

PyTorch

from torchvision import datasets, transforms

transform = transforms.ToTensor()
cifar10_train = datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)

Load FashionMNIST training dataset, which has clothes images.

PyTorch

from torchvision import datasets

fashion_train = datasets.FashionMNIST(root='./data', train=True, download=True)

Sample Model

This code loads the MNIST training data, converts images to tensors, and prints the shape of one batch of images and labels along with the first 10 labels.

PyTorch

import torch
from torchvision import datasets, transforms
from torch.utils.data import DataLoader

# Define a transform to convert images to tensors
transform = transforms.ToTensor()

# Load MNIST training dataset
mnist_train = datasets.MNIST(root='./data', train=True, download=True, transform=transform)

# Create a data loader to iterate over dataset
train_loader = DataLoader(mnist_train, batch_size=64, shuffle=True)

# Get one batch of images and labels
images, labels = next(iter(train_loader))

print(f'Batch image tensor shape: {images.shape}')
print(f'Batch label tensor shape: {labels.shape}')
print(f'First 10 labels in batch: {labels[:10].tolist()}')

Important Notes

Built-in datasets automatically download data if not found locally.

Transforms help convert raw data into a format your model can use.

Use DataLoader to easily batch and shuffle data for training.

Summary

Built-in datasets in torchvision provide easy access to popular image datasets.

They save time by handling download and loading for you.

Use transforms and DataLoader to prepare data for your model.

Practice

1. What is the main purpose of using torchvision.datasets in PyTorch?

easy

A. To visualize model training progress

B. To create custom neural network layers

C. To easily download and load popular image datasets

D. To optimize model hyperparameters automatically

Solution

1. Step 1: Understand the role of torchvision.datasets

   It provides built-in access to popular image datasets like CIFAR10, MNIST, etc., handling download and loading.

2. Step 2: Compare with other options

   Options A, B, and D relate to different PyTorch functionalities, not dataset loading.

3. Final Answer:

   To easily download and load popular image datasets -> Option C

4. Quick Check:

   torchvision.datasets = dataset loading [OK]

Hint: Remember: torchvision.datasets = easy dataset access [OK]

Common Mistakes:

• Confusing datasets with model layers
• Thinking it handles training visualization
• Assuming it tunes hyperparameters

2. Which of the following is the correct way to import the CIFAR10 dataset from torchvision?

easy

A. from torchvision.datasets import CIFAR10

B. import torchvision.datasets.CIFAR10

C. from torchvision import CIFAR10

D. import CIFAR10 from torchvision.datasets

Solution

1. Step 1: Recall Python import syntax for modules and classes

   The correct syntax to import a class from a module is: from module import Class.

2. Step 2: Apply to torchvision.datasets and CIFAR10

   So, from torchvision.datasets import CIFAR10 is correct. Other options have syntax errors or wrong order.

3. Final Answer:

   from torchvision.datasets import CIFAR10 -> Option A

4. Quick Check:

   Correct import syntax = from torchvision.datasets import CIFAR10 [OK]

Hint: Use 'from module import Class' for dataset imports [OK]

Common Mistakes:

• Using dot notation in import statement incorrectly
• Swapping import order
• Trying to import dataset directly without module

3. What will be the output shape of images loaded from the MNIST dataset using torchvision.datasets with default settings?

medium

A. (28, 28)

B. (1, 28, 28)

C. (3, 28, 28)

D. (32, 32, 3)

Solution

1. Step 1: Recall MNIST image format

   MNIST images are grayscale with size 28x28 pixels.

2. Step 2: Understand PyTorch tensor shape for images

   PyTorch uses channel-first format: (channels, height, width). Grayscale images have 1 channel, so shape is (1, 28, 28).

3. Final Answer:

   (1, 28, 28) -> Option B

4. Quick Check:

   MNIST images = grayscale 28x28 = (1,28,28) [OK]

Hint: PyTorch images: channels first, grayscale = 1 channel [OK]

Common Mistakes:

• Assuming shape is height x width only
• Confusing RGB (3 channels) with grayscale
• Using (28, 28) without channel dimension

4. Consider this code snippet:

from torchvision.datasets import CIFAR10
train_data = CIFAR10(root='./data', train=True, download=True)
print(len(train_data))

What error will occur if download=True is omitted and data is not already downloaded?

medium

A. SyntaxError: Missing download argument

B. No error, data loads automatically

C. FileNotFoundError: No such file or directory

D. RuntimeError: Dataset not found and download=False

Solution

1. Step 1: Understand CIFAR10 dataset download behavior

   If download=False (default) and data is missing, it raises an error because it won't download automatically.

2. Step 2: Identify the error type

   The error raised is a RuntimeError indicating dataset not found and download flag is False.

3. Final Answer:

   RuntimeError: Dataset not found and download=False -> Option D

4. Quick Check:

   Missing data + download=False = RuntimeError [OK]

Hint: Always set download=True first time to avoid RuntimeError [OK]

Common Mistakes:

• Expecting automatic download without flag
• Confusing RuntimeError with SyntaxError
• Assuming FileNotFoundError is raised

5. You want to load the CIFAR10 training dataset with images normalized to mean=0.5 and std=0.5 for each RGB channel. Which code snippet correctly applies this transform using torchvision?

hard

A. from torchvision.datasets import CIFAR10
from torchvision.transforms import Compose, ToTensor, Normalize
transform = Compose([ToTensor(), Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))])
data = CIFAR10(root='./data', train=True, download=True, transform=transform)

B. from torchvision.datasets import CIFAR10
transform = Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))
data = CIFAR10(root='./data', train=True, download=True, transform=transform)

C. from torchvision.datasets import CIFAR10
from torchvision.transforms import ToTensor
transform = ToTensor()
data = CIFAR10(root='./data', train=True, download=True, transform=transform)
data = Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))

D. from torchvision.datasets import CIFAR10
from torchvision.transforms import Compose, Normalize
transform = Compose([Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))])
data = CIFAR10(root='./data', train=True, download=True, transform=transform)

Solution

1. Step 1: Understand required transforms for normalization

   Images must first be converted to tensors with ToTensor() before normalization.

2. Step 2: Check transform composition

   Use Compose to chain ToTensor() and Normalize() in order.

3. Step 3: Verify each option

   from torchvision.datasets import CIFAR10 from torchvision.transforms import Compose, ToTensor, Normalize transform = Compose([ToTensor(), Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))]) data = CIFAR10(root='./data', train=True, download=True, transform=transform) correctly composes both transforms and passes to dataset. Others miss ToTensor() or apply transforms incorrectly.

4. Final Answer:

   Option A code snippet with Compose, ToTensor, Normalize -> Option A

5. Quick Check:

   Normalize after ToTensor using Compose = from torchvision.datasets import CIFAR10 from torchvision.transforms import Compose, ToTensor, Normalize transform = Compose([ToTensor(), Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))]) data = CIFAR10(root='./data', train=True, download=True, transform=transform) [OK]

Hint: Always use ToTensor before Normalize in Compose [OK]

Common Mistakes:

• Applying Normalize without ToTensor first
• Not using Compose to chain transforms
• Applying transforms outside dataset argument