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