DatasetFolder

类 torchvision.datasets.DatasetFolder(root: Union[str, Path], loader: Callable[[str], Any], extensions: Optional[Tuple[str, ...]] = None, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, is_valid_file: Optional[Callable[[str], bool]] = None, allow_empty: bool = False)

一个通用的数据加载器。

可以通过重写 find_classes() 方法来定制这种默认的目录结构。

参数：

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

• loader (callable) – 给定路径加载样本的函数。

• extensions (tuple[string]) – 允许的扩展名列表。extensions 和 is_valid_file 不应同时传递。

• transform (callable, optional) – 一个函数/变换，它接收样本并返回变换后的版本。例如，用于图像的 transforms.RandomCrop。

• target_transform (callable, optional) – 一个函数/变换，它接收目标并对其进行变换。

• is_valid_file (callable, optional) – 一个函数，它接收文件路径并检查文件是否有效（用于检查损坏的文件）。extensions 和 is_valid_file 不应同时传递。

• allow_empty – 如果为 True，空文件夹被视为有效类别。如果为 False（默认值），则在空文件夹上引发错误。

find_classes(directory: Union[str, Path]) → Tuple[List[str], Dict[str, int]]

在如下结构的数据集中查找类别文件夹

directory/
├── class_x
│   ├── xxx.ext
│   ├── xxy.ext
│   └── ...
│       └── xxz.ext
└── class_y
    ├── 123.ext
    ├── nsdf3.ext
    └── ...
    └── asd932_.ext

可以重写此方法以仅考虑类别的子集，或适应不同的数据集目录结构。

参数：

directory (str) – 根目录路径，对应于 self.root

引发：

FileNotFoundError – 如果 dir 没有类别文件夹。

返回：

所有类别的列表以及将每个类别映射到索引的字典。

返回类型：

(Tuple[List[str], Dict[str, int]])

静态 make_dataset(directory: Union[str, Path], class_to_idx: Dict[str, int], extensions: Optional[Tuple[str, ...]] = None, is_valid_file: Optional[Callable[[str], bool]] = None, allow_empty: bool = False) → List[Tuple[str, int]]

生成形式为 (path_to_sample, class) 的样本列表。

可以重写此方法，例如，从压缩的 zip 文件而不是从磁盘读取文件。

参数：

• directory (str) – 数据集根目录，对应于 self.root。

• class_to_idx (Dict[str, int]) – 将类别名称映射到类别索引的字典。

• extensions (optional) – 允许的扩展名列表。应传递 extensions 或 is_valid_file 中的一个。默认为 None。

• is_valid_file (optional) – 一个函数，它接收文件路径并检查文件是否有效（用于检查损坏的文件）。extensions 和 is_valid_file 不应同时传递。默认为 None。

• allow_empty (bool, optional) – 如果为 True，空文件夹被视为有效类别。如果为 False（默认值），则在空文件夹上引发错误。

引发：

• ValueError – 如果 class_to_idx 为空。

• ValueError – 如果 extensions 和 is_valid_file 均为 None 或均不为 None。

• FileNotFoundError – 如果任何类别都没有找到有效文件。

返回：

形式为 (path_to_sample, class) 的样本

返回类型：

List[Tuple[str, int]]