是的,可以使用datasets.ImageFolder函数来加载没有文件夹结构的图像数据集。datasets.ImageFolder函数会默认将 root 目录下的子文件夹作为不同类别的图像数据,但如果 root 目录下只有图片文件而没有文件夹,则所有图片会被当作同一类别处理。你可以使用以下代码来加载这种类型的数据集:

data_dir = 'G:/image' 
dataset_val = datasets.ImageFolder(root=data_dir, transform=transform)

这样会将 G:/image 目录下的所有图片都加载到 dataset_val 中。

data_dir = Gimage dataset_val = datasetsImageFolderroot=data_dir transform=transformimage目录下没有文件夹了全部是图片这样构建可以吗?

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

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