pytorch - 通过torch DataLoader加载微型图像网的问题

Question

我正在使用 tiny-imagenet-200，但不确定是否可以使用 torch.utils.data.DataLoader 加载它们。我从斯坦福网站下载了 tiny-imagenet-200，但验证的格式设置在名为 val_0 到 val_9999 的目录中，它们的标签在 .txt 中。如何通过 torch.utils.data.DataLoader 加载此目录？我试过了：

datasets.ImageFolder(args.val_dir, transforms.Compose([
            OpencvResize(256),
            transforms.CenterCrop(224),
            ToBGRTensor(),
        ]) 

但它不起作用。

标签： pytorchtorchtorchvisionimagenet

Answer 1

你不能ImageFolder直接使用。不过，还有其他选择：

• 您可以阅读注释文件并重新构建目录以启用 的使用ImageFolder，如此处；
• 您可以实现自定义Dataset. 幸运的是，由于 Tiny ImageNet 是一个流行的数据集，您可以在网上找到许多实现。比如这个。