Pytorch torchvision.datasets.DatasetFolder.find_classes()

from torchvision.datasets.DatasetFolder import find_classes
find_classes(directory(str))

参数

directory(str): 文件路径

输出

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

输出类型

(元组[List[str], Dict[str, int]])

文件夹类型

directory/
├── class_1
│   ├── xxx.ext
│   ├── xxy.ext
│   └── ...
│       └── xxz.ext
└── class_2
    ├── 123.ext
    ├── 3.ext
    └── ...
    └── 2_.ext