PyTorch读取Tiny-ImageNet-200验证集的代码:
import torch
import torchvision.datasets as datasets
import torchvision.transforms as transforms
# 定义数据增强和标准化转换
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
val_dataset = datasets.ImageFolder(root='path/to/tiny-imagenet-200/val', transform=transform)
# 创建数据加载器,批次大小为32
val_loader = torch.utils.data.DataLoader(val_dataset, batch_size=32, shuffle=False)
请确保将path/to/tiny-imagenet-200/val
替换为您Tiny-ImageNet-200验证集的实际路径。