通过torch DataLoader加载微型imagenet的问题

Question

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

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

但不起作用。

Answer 1

您不能直接使用ImageFolder来做到这一点。不过，也有其他选择：

• 您可以读取注释文件并重新构造目录以启用ImageFolder的使用，就像在这里中一样；
• 您可以实现自定义Dataset。幸运的是，由于微型ImageNet是一个流行的数据集，您可以在网上找到许多实现。例如，这一个。