AttributeError: module 'torchvision.datasets' has no attribute 'FashionMNIST'
时间: 2023-10-28 13:59:04 浏览: 211
这个错误是因为您正在尝试使用 Torchvision 库的 FashionMNIST 数据集,但是 Torchvision 不支持 FashionMNIST 数据集。Torchvision 仅支持 MNIST,CIFAR10,CIFAR100 和 ImageNet 等数据集。
如果您想使用 FashionMNIST 数据集,您可以直接从官方网站下载并加载数据集,然后使用 Torch 的数据加载器进行处理和训练。以下是一个示例代码:
```python
import torchvision.transforms as transforms
from torch.utils.data import DataLoader
from torchvision.datasets import DatasetFolder
# 设置数据集路径和转换
data_path = 'path_to_fashionmnist_folder'
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,))
])
# 加载训练集和测试集
train_dataset = DatasetFolder(
root=data_path,
loader=torchvision.datasets.ImageFolder.default_loader,
extensions='.png',
transform=transform
)
test_dataset = DatasetFolder(
root=data_path,
loader=torchvision.datasets.ImageFolder.default_loader,
extensions='.png',
transform=transform
)
# 创建数据加载器
train_loader = DataLoader(train_dataset, batch_size=64, shuffle=True)
test_loader = DataLoader(test_dataset, batch_size=64, shuffle=False)
```
请确保替换 `data_path` 变量为您下载的 FashionMNIST 数据集的路径,然后您就可以使用 `train_loader` 和 `test_loader` 进行训练和测试了。
阅读全文