如何使用 datasets.ImageFolder 获取图片路径
datasets.ImageFolder 通过传递图片文件夹的路径来自动获取每张图片的路径。在使用 ImageFolder 时,可以指定图片文件夹所在的路径,例如:
data_dir = './data'
image_dataset = datasets.ImageFolder(root=data_dir)
这里的 data_dir 是包含图片文件夹的路径,ImageFolder 会自动扫描该路径下的所有子文件夹,每张图片的路径会被自动获取。可以通过 image_dataset.samples 属性来查看每张图片的路径和对应的类别,例如:
print(image_dataset.samples)
输出的结果类似于:
[('./data/class1/image1.jpg', 0), ('./data/class1/image2.jpg', 0), ('./data/class2/image1.jpg', 1), ('./data/class2/image2.jpg', 1), ...]
其中每个元素的第一个元素是图片的路径,第二个元素是类别的标签。
原文地址: https://www.cveoy.top/t/topic/lFsZ 著作权归作者所有。请勿转载和采集!