torch.utils.data和torvision等PyTorch数据处理工具箱

数据处理工具箱概述 Pytorch涉及数据处理（数据装载，数据预处理，数据增强等）主要工具包及相互关系如图所示： 图的左边是torch.utils.data工具包，主要包括以下4个类： Dataset：是一个抽象类，其他数据集要继承这个类，并且重写其中两个方法（getitem, len）。 DataLoader：定义一个新的迭代器，实现批量（batch）读取，打乱数据（shuffle）并提供并行加

>>阅读原文<<