pytorch：获取给定 ImageFolder 数据集的类数

Question

如果我有一个像这样的数据集：image_datasets['train'] = datasets.ImageFolder(train_dir, transform=train_transforms)如何以编程方式确定数据集中的类或唯一标签的数量？

Answer 1

用：

len(image_datasets['train'].classes)

.classes 返回一个列表。

Answer 2

如果您的数据类型是张量，则可以使用：

import torch

 n_classes = len(torch.unique(Your_Target_Vector))