如何在pytorch中从DataFolder.Samples系统地获取数据?

时间:2018-10-31 14:08:38

标签: python pytorch torchvision

我正在使用AT&T人脸数据集。有40个文件夹,每个文件夹包含一个人的10张图像。 我可以使用以下方法轻松获得随机类别的图像:

d = datasets.ImageFolder('./data/')
data = random.choice(d.samples)
label = data[1]

现在,我需要将此图像与另一个类别的第(i)个样本图像(例如第34个)配对,但是在尝试从样本访问特定图像时不起作用。我知道我们可以随机遍历样本并检查是否找到了所需的类图像,但是该方法似乎效率不高。

那么,有没有一种方法可以手动访问特定类别的图像,而不是随机选择图像?

0 个答案:

没有答案