Skip to content

Instantly share code, notes, and snippets.

View isaackhlam's full-sized avatar

CommitSudoku isaackhlam

View GitHub Profile
@isaackhlam
isaackhlam / Select_CIFAR10_Classes.py
Created October 11, 2023 15:21 — forked from Miladiouss/Select_CIFAR10_Classes.py
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()