假设我正在使用以下调用:
trainset = torchvision.datasets.ImageFolder(root="imgs/", transform=transform)
trainloader = torch.utils.data.DataLoader(trainset,batch_size=4,shuffle=True,num_workers=1)
Run Code Online (Sandbox Code Playgroud)
据我所知,这将 定义trainset
为由文件夹“images”中的所有图像组成,并具有由特定文件夹位置定义的标签。
我的问题是 - 是否有任何直接/简单的方法可以将 定义trainset
为该文件夹中图像的子样本?例如,定义trainset
为每个子文件夹中 10 张图像的随机样本?