Amish Goel agoel10

Miladiouss / Select_CIFAR10_Classes.py

Last active October 27, 2023 03:41

Create PyTorch datasets and dataset loaders for a subset of CIFAR10 classes.

	import torchvision
	import torchvision.transforms as transforms
	from torchvision.datasets import CIFAR10
	from torch.utils.data import Dataset, DataLoader
	import numpy as np

	# Transformations
	RC = transforms.RandomCrop(32, padding=4)
	RHF = transforms.RandomHorizontalFlip()
	RVF = transforms.RandomVerticalFlip()