Pytorch Note52 灵活的数据读取介绍
全部笔记的汇总贴: Pytorch Note 快乐星球
图片数据一般有两种情况:
1、所有图片放在一个文件夹内,另外有一个txt文件显示标签。
2、不同类别的图片放在不同的文件夹内,文件夹就是图片的类别。
针对这两种不同的情况,数据集的准备也不相同,第一种情况可以自定义一个Dataset,第二种情况直接调用torchvision.datasets.ImageFolder
来处理。下面分别进行说明:
灵活的数据读取
首先导入我们需要的函数
from torchvision.datasets import ImageFolder
文件中数据分布是这样的,每个文件夹中有三张图片