Python PyTorch DatasetFolder.find_classes用法及代码示例

本文简要介绍python语言中 torchvision.datasets.DatasetFolder.find_classes 的用法。

用法:

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

参数：

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

抛出：

FileNotFoundError - 如果dir 没有类文件夹。

返回：

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

返回类型：

(元组[列表[str]，字典[str，int]])

在数据集中查找类文件夹，其结构如下：

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

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