torch.utils.data.DataLoader 官方文档学习
阅读原文时间:2021年04月20日阅读:1

Pytorch 首选的数据上传工具是 torch.utils.data.DataLoader,它可以将已经预处理完成的数据集转换成一个迭代器,迭代一次输出一个预先定义好的batch_size数量的图片

torch.utils.data.DataLoader类

1、**torch.utils.data.DataLoader**(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, multiprocessing_context=None)

Dataloader类结合数据集和采样器,同时提供包含传入数据集的迭代器

它同时支持map_style和iterable_style类型的数据集,同时支持单进程和多进程加载数据,及自定义的加载顺序,可batch_size

2、参数详解

dataset: 将要加载的数据的数据集

batch_size(int): 每迭代加载的图像的数量

shuffle(bool):将其设置为True时,数据将会在每个epoch之后,进行打乱重新排序

sampler(Sampler): 定义在数据集中抓取样本的方法

batch_sampler(Sampler):同sampler,同时还会返回这一批样本的索引

num_workers(int):定义家在数据的线程数,默认为0,意味着只使用主线程加载数据。

。。。

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章