PyTorch提供了ImageFolder的类来加载文件结构如下的图片数据集:
root/dog/xxx.png
root/dog/xxy.png
root/dog/xxz.png
root/cat/123.png
root/cat/nsdf3.png
root/cat/asd932_.png
使用这个类的问题在于无法将训练集(training dataset)和验证集(validation dataset)分开。我写了两个类来完成这个工作。
import os
import torch
from torch.utils.data import Dataset, DataLoader
f
1