pytorch加载自己的数据集图片格式

适用于加载文件夹中的图片

dset.ImageFolder(root="root folder path", [transform, target_transform])

他有以下成员变量:

• self.classes - 用一个list保存 类名
• self.class_to_idx - 类名对应的 索引
• self.imgs - 保存(img-path, class) tuple的list

例子

traindir = os.path.join(args.data, 'train')
	valdir = os.path.join(args.data, 'val')
	train = datasets.ImageFolder(traindir, transform)
	val = datasets.ImageFolder(valdir, transform)
	train_loader = torch.utils.data.DataLoader(
	train, batch_size=args.batchSize, shuffle=True, num_workers=args.nThreads)