datasets.ImageFolder是什么意思
时间: 2023-05-17 07:04:47 浏览: 201
pytorch torchvision.ImageFolder的用法介绍
datasets.ImageFolder是PyTorch中的一个类,用于加载图像数据集。它假定图像数据集的文件夹结构如下:每个类别的图像都存储在一个单独的文件夹中,文件夹的名称即为类别的名称。该类可以自动地将图像转换为PyTorch中的张量,并提供了一些方便的方法来访问和操作数据集。
阅读全文
pytorch torchvision.ImageFolder的用法介绍