小编Dr.*_*ohn的帖子

如何让火车装载机使用特定数量的图像?

假设我正在使用以下调用:

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 张图像的随机样本?

python pytorch dataloader

4
推荐指数
1
解决办法
3429
查看次数

标签 统计

dataloader ×1

python ×1

pytorch ×1