使用 datasets.ImageFolder 获取特定图片路径

可以通过datasets.ImageFolder的imgs属性获取所有图片的路径和标签，然后根据需要找到指定图片的路径。例如：

from torchvision import datasets

# 创建 ImageFolder 数据集
data_dir = '/path/to/your/data'
dataset = datasets.ImageFolder(data_dir)

# 获取所有图片的路径和标签
imgs = dataset.imgs

# 找到指定图片的路径
img_path = None
for path, label in imgs:
    if 'target_img.jpg' in path:
        img_path = path
        break

# 显示指定图片
if img_path is not None:
    from PIL import Image
    img = Image.open(img_path)
    img.show()
else:
    print('Target image not found.')

其中，data_dir是数据集的根目录，'target_img.jpg'是要查找的图片文件名，可以根据实际情况修改。img_path最终存储的是找到的图片路径，如果没有找到则为None。最后，如果找到了指定图片，则可以使用PIL库读取并显示图片。