pytorch制作自己的数据集

pytorch制作自己的数据集

datasets的Imagefolder模块

文件格式如图

transformer = transforms.Compose([transforms.Resize([224,224]),
                                                  transforms.ToTensor()])
trainsets = datasets.ImageFolder('./train_dataset',transformer)
testsets = datasets.ImageFolder('./test_dataset',transformer)

datasets就做好了
类似datasets下载MINIST数据集的功能
下载后，可以用

所以可以用dataloader进行批处理方便后续将数据放入模型进行训练

trainloader = torch.utils.data.DataLoader(trainsets, batch_size=1, shuffle=True, num_workers=4)
testloader = torch.utils.data.DataLoader(testsets, batch_size=1, shuffle=True, num_workers=4)