pytorch加载本地cifar数据集

下面是一个使用PyTorch加载本地CIFAR数据集的代码示例：

import torch
from torchvision import datasets, transforms

# 数据预处理的转换
transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
])

# CIFAR-10数据集的根目录
data_dir = './cifar_data'

# 加载训练集和测试集
train_dataset = datasets.CIFAR10(root=data_dir, train=True, download=False, transform=transform)
test_dataset = datasets.CIFAR10(root=data_dir, train=False, download=False, transform=transform)

# 创建数据加载器
train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=64, shuffle=True)
test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=64, shuffle=False)

# 可以使用train_loader和test_loader来迭代获取数据
for images, labels in train_loader:
    # 在这里进行训练数据的处理
    pass

for images, labels in test_loader:
    # 在这里进行测试数据的处理
    pass

上述代码中，首先定义了数据预处理的转换，将图像转换为张量并进行归一化。然后指定了CIFAR-10数据集的根目录，这个目录应该包含名为`train`和`test`的子文件夹，分别存放训练集和测试集的图像文件。接下来使用`datasets.CIFAR10`类加载训练集和测试集，指定数据集的根目录、训练集/测试集、是否下载以及数据预处理转换。最后使用`torch.utils.data.DataLoader`创建数据加载器，指定数据集和批次大小。

请确保将`data_dir`变量设置为正确的CIFAR-10数据集的根目录路径，并确保数据集已经下载到该路径下。