pytorch学习笔记:dataloader与dataset

pytorch学习笔记:dataloader与dataset_第1张图片
sampler:生成索引
dataset:根据索引读取图片及标签

Dataloader

pytorch学习笔记:dataloader与dataset_第2张图片
Epoch:所有训练样本都输入到模型中,成为一个epoch
iteration:一批样本输入到模型中,称之为一个iterration
batchsize:批大小,决定一个epoch有多少iteration

Dataset

pytorch学习笔记:dataloader与dataset_第3张图片pytorch学习笔记:dataloader与dataset_第4张图片
pytorch学习笔记:dataloader与dataset_第5张图片加入transform后的流程图
pytorch学习笔记:dataloader与dataset_第6张图片
sampler:选出读取数据的index
dataset:设置读取数据的目录
getitem:定义如何读取数据

transforms.normalize
功能:逐通道对图像进行标准化
参数:mean,std,inplace是否原地操作

本文为深度之眼Pytorch课程的学习笔记,仅供自己学习使用,如有问题欢迎讨论!关于课程可以扫描下图二维码

pytorch学习笔记:dataloader与dataset_第7张图片

你可能感兴趣的:(pytorch学习笔记)