pytorch学习笔记七：torchvision.datasets.ImageFolder使用详解

一、数据集组织方式

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

对于上面的root，假设data文件夹在.py文件的同级目录中，那么root一般都是如下这种形式：./data/train和./data/valid

二、ImageFolder参数详解

dataset=torchvision.datasets.ImageFolder(
                       root, transform=None