有没有自动下载CIFAR-10数据集的代码
时间: 2024-12-16 14:10:27 浏览: 5
CIFAR-10是一个常用的小型图像识别数据集,用于训练机器学习模型。如果你想在Python中自动下载这个数据集,可以使用像`torchvision.datasets`这样的库,它已经包含了对CIFAR-10的支持。以下是一个简单的例子:
```python
import torch
from torchvision import datasets, transforms
# 定义数据预处理步骤
data_transforms = {
'train': transforms.Compose([
transforms.RandomHorizontalFlip(),
transforms.RandomCrop(32, padding=4),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
]),
'val': transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])
}
# 下载并加载CIFAR-10数据集
train_dataset = datasets.CIFAR10(root='./data', train=True, download=True, transform=data_transforms['train'])
test_dataset = datasets.CIFAR10(root='./data', train=False, download=True, transform=data_transforms['val'])
```
这段代码首先导入必要的库,然后定义了数据预处理流程。`download=True`参数会检查数据是否已存在,如果不存在则自动下载。
阅读全文