【Pytorch实战教程】torchvision.datasets.ImageFolder的详细用法介绍
下面通过一个完整的实例详细讲解 torchvision.datasets.ImageFolder
的用法,包括数据准备、加载、预处理和可视化全过程:
它的主要作用是直接从文件夹下读取图片数据,不需要逐个读取再拼接为训练集或测试集了
场景设定:垃圾分类数据集
假设我们有如下目录结构的垃圾分类数据集:
garbage_classification/├── train/│ ├── paper/│ │ ├── paper001.jpg│ │ └── paper002.jpg│ ├── plastic/│ │ ├── plastic001.jpg│ │ └── plastic002.jpg│ └── glass/│ ├── glass001.jpg│ └── glass002.jpg└── test/ ├── paper/ │ └── paper003.jpg ├── plastic/ │ └── plastic003.jpg └── glass/ └── glass003.jpg
步骤1:导入必要库