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

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)