20天狂宴Pytorch-Day2
图片数据建模.
准备数据
cifar2 数据集包含 5000 张飞机 airplane 照片和 5000 张机动车 automobile 照片, 任务目标是训练一个模型区分 airplane 和 automobile 两种图片.
在 Pytorch 中构建图片数据管道通常有两种方法:
- 使用 torchvision 中的 datasets.ImageFolder 读取图片, 然后用 DataLoader 并行加载
- 通过继承 torch.utils.data.Dataset 实现用户自定义读取逻辑, 然后用 DataLoader 并行加载
本文介绍第一种方法.