datasets.imagefolder怎么获取某张图片的路径

在使用datasets.ImageFolder时，可以通过dataset.samples属性获取所有图片的路径和标签。其中，每个元素是一个元组，包含图片路径和对应的标签。例如：

import torchvision.datasets as datasets

data_dir = 'path/to/your/dataset'
dataset = datasets.ImageFolder(root=data_dir)

# 获取第一张图片的路径和标签
img_path, label = dataset.samples[0]
print(img_path)

输出：

path/to/your/dataset/class1/image1.jpg

如果要获取某个特定标签下的所有图片路径，可以使用dataset.class_to_idx属性获取标签和对应的索引，然后遍历dataset.samples，筛选出特定标签的图片路径。例如：

# 获取标签为'class1'的所有图片路径
class1_idx = dataset.class_to_idx['class1']
class1_img_paths = [path for path, label in dataset.samples if label == class1_idx]
print(class1_img_paths)

输出：

['path/to/your/dataset/class1/image1.jpg', 'path/to/your/dataset/class1/image2.jpg', ...]