一、PyTorch 加载数据

1. Dataset

提供一种方式去获取数据及其label

2. Dataloader

为后面的网络提供不同的数据形式

二、OS

  • listdir函数

image.png

  • os.path.join函数

image.png

三、 实战

image.png

  1. getitem_ 和为 len重写的方法。

gititem中idx为参数,函数返回图片和图片的标签
len函数返回数据集的个数