Pytorch数据处理工具箱
·

数据处理工具箱概述

utils.data
_getitem_一次只能获取一个样本。
1.utils.data.Dataset
2.DataLoader:可以批量处理。
transforms
transforms提供了对PIL Image对象和Tensor对象的常用操作。
如果要对数据集进行多个操作,可通过Compose将这些操作像管道一样拼接起来,类似于nn.Sequential。
ImageFolder
ImageFolder可以读取不同目录下的图像数据。
更多推荐
所有评论(0)