Pytorch数据处理工具箱

Pytorch数据处理工具箱
数据处理工具箱概述

utils.data

·_getitem_一次只能获取一个样本。

utils.data
1.utils.data.Dataset

utils.data
2.DataLoader：可以批量处理。
  语法结构如右所示。

data.DataLoader(
    dataset,
    batch_size=1,
    shuffle=False,
    sampler=None,
    batch_sampler=None,
    num_workers=0,
    collate_fn=,
    pin_memory=False,
    drop_last=False,
    timeout=0,
    worker_init_fn=None,
)
utils.data
2.DataLoader：可以批量处理。相关参数介绍如下所示。 

utils.data
2.DataLoader：可以批量处理。
但是！数据处在不同目录之下时会不方便！
但是DataLoader并不是迭代器，
可以通过“iter”命令转换成迭代器。

 

 

torchvision