Pytorch框架学习（5）——DataLoder与Dataset

Dataloder与DataSet

1. 人民币二分类

今天主要分享机器学习训练过程的数据处理过程。

• 数据：
  • 数据收集：Img，Label
  • 数据划分：训练集，验证集，测试集
  • 数据读取：DataLoder（Sampler：生成索引，DataSet：根据索引读取图片Img和标签Label）
  • 数据预处理：transforms
    

2. DataLoder与Dataset

2.1 DataLoder

• torch.utils.data.DataLoder

• 功能：构建可迭代的数据装载器，训练时，每一个for循环就是从DataLoder中获取一个batchsize大小的数据

  • dataset：Dataset类，决定数据从哪读取及如何读取
  • batchsize：批大小
  • num_works