从 PyTorch 中的数据加载器打印随机样本

【问题标题】：Print random sample from dataloader in PyTorch从 PyTorch 中的数据加载器打印随机样本
【发布时间】：2020-11-23 18:33:30
【问题描述】：

我将某个数据集加载到数据加载器中。例如，如果我想从这个数据加载器中保存 100 张图像，我应该如何遍历数据加载器来保存它们？

【问题讨论】：

嗨@CaxiomX20，欢迎来到SO！你都尝试了些什么？请提供更多信息来帮助我们

标签： python dataset pytorch dataloader

【解决方案1】：

我不确定您要做什么（也许编辑您的问题），但也许这会有所帮助：

dataset = Dataset()
dataloader = torch.utils.data.DataLoader(
                 dataloader,
                 batch_size=32,
                 num_workers=1,
                 shuffle=True)

for samples, targets in dataloader:
    # 'sample' now is a batch of 32 (see batch-size above) elements of your dataset

这是你想要的吗？希望如此:)

【讨论】：