m0_49704897的博客

原创 PyTorch数据加载全攻略

​Data Set是一个抽象类，表示一个数据集。__init__: 在 实例化DataSet 对象运行一次。我们初始化包含图像的目录、注释文件和transform与 target_transform.__len__：返回数据集的总样本数。会调用它。：根据整数索引idx会返回一个样本（通常为特征和标签）。会调用它。其作用就是实现通过索引访问对应的数据以及标签。使用自定义数据集时，可以用将其与结合使用，以便进行数据的批量加载和处理和训练。映射式（Map-Style）和可迭代式（Iterable-style）