Pytorch torchvision.datasets.DatasetFolder.find

torchvision.datasets.DatasetFolder.find"/>

Pytorch torchvision.datasets.DatasetFolder.find

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