文章目录 1.自带的datasets2.ImageFolder3.自己创建的dataset(用的多) 1.自带的datasets pytorch自带的数据集 具体数据集如下: 用法: 可以用的功能 2.ImageFolder 文件形式如下时用此:(一个文件下只有一个类别) 例如:天气数据集data 下有 train 跟test train下有四个类别(多云、晴、雨、雪) test下有四个类别 例如 多云文件下 全是多云的图片 后面的用法跟pytorch自带数据集的用法一样。 3.自己创建的dataset(用的多) 可以根据的需求,随意添加 代码如下: