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)