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

可以重寫此方法以僅考慮類的子集，或適應不同的數據集目錄結構。