快捷方式

ImageFolder

class torchvision.datasets.ImageFolder(root: ~typing.Union[str, ~pathlib.Path], transform: ~typing.Optional[~typing.Callable] = None, target_transform: ~typing.Optional[~typing.Callable] = None, loader: ~typing.Callable[[str], ~typing.Any] = <function default_loader>, is_valid_file: ~typing.Optional[~typing.Callable[[str], bool]] = None, allow_empty: bool = False)[源代码]

一个通用的数据加载器,其中图像默认情况下按这种方式排列

root/dog/xxx.png
root/dog/xxy.png
root/dog/[...]/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/[...]/asd932_.png

此类继承自 DatasetFolder,因此可以覆盖相同的方法来自定义数据集。

参数:
  • **root** (str 或 pathlib.Path) – 根目录路径。

  • **transform** (可调用可选) – 一个函数/转换,它接受一个 PIL 图像并返回一个转换后的版本。例如,transforms.RandomCrop

  • **target_transform** (可调用可选) – 一个函数/转换,它接受目标并对其进行转换。

  • **loader** (可调用可选) – 一个函数,用于根据其路径加载图像。

  • **is_valid_file** (可调用可选) – 一个函数,它接受图像文件的路径并检查该文件是否为有效文件(用于检查损坏的文件)

  • **allow_empty** – 如果为 True,则空文件夹被视为有效类别。如果为 False(默认),则在空文件夹上会引发错误。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源