【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：导入必要库