如果你的目录下只有图片而没有文件夹,你可以将图片放在一个文件夹内,然后将该文件夹作为数据集的根目录。例如,你可以创建一个名为"images"的文件夹,将所有的图片放在该文件夹内,然后将"images"作为数据集的根目录。\n\n这样,你可以按照以下方式构建数据集:\n\npython\ndata_dir = \"G:/flickr_pic/images\" # 图片所在的文件夹\ndataset_val = datasets.ImageFolder(root=data_dir, transform=transform)\n\n\n请确保在指定数据集根目录时,将文件夹的路径正确地传递给datasets.ImageFolderroot参数。


原文地址: https://www.cveoy.top/t/topic/p67F 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录