machine-learning - 如何在 Pytorch 上加载 Omniglot

Question

我正在尝试在 Omniglot 数据集上做一些实验，我看到 Pytorch 实现了它。我已经运行了命令

from torchvision.datasets import Omniglot

但我不知道如何实际加载数据集。有没有一种方法可以像我们打开MNIST一样打开它？类似于以下内容:

train_dataset = dsets.MNIST(root='./data', 
                            train=True, 
                            transform=transforms.ToTensor(),
                            download=True)

最终目标是能够分别打开训练集和测试集并对其进行实验。

Answer 1

您可以执行与 Omniglot 包含 图像 和 标签 完全相同的转换，就像 MNIST 一样，例如:

import torchvision

dataset = torchvision.datasets.Omniglot(
    root="./data", download=True, transform=torchvision.transforms.ToTensor()
)

image, label = dataset[0]
print(type(image))  # torch.Tensor
print(type(label))  # int