我正在使用AT&T人脸数据集。有40个文件夹,每个文件夹包含一个人的10张图像。 我可以使用以下方法轻松获得随机类别的图像:
d = datasets.ImageFolder('./data/')
data = random.choice(d.samples)
label = data[1]
现在,我需要将此图像与另一个类别的第(i)个样本图像(例如第34个)配对,但是在尝试从样本访问特定图像时不起作用。我知道我们可以随机遍历样本并检查是否找到了所需的类图像,但是该方法似乎效率不高。
那么,有没有一种方法可以手动访问特定类别的图像,而不是随机选择图像?