数据集预处理
多分类数据集的dataset操作
- 对于文件夹名字是训练模型所需label时,如下图
我们可以通过使用torchvision.datasets.ImageFolder来对我们的多个dataset进行数据加载
train_transforms_dataset = torchvision.datasets.ImageFolder(root=train_path, transform=transforms_img)
加载完成后即可使用DataLoader进行数据分批操作
train_transforms_loader = DataLoader(train_transforms_dataset, batch_size=64, shuffle=True, drop_last=False)