如果我有一个如下所示的数据集：

image_datasets['train'] = datasets.ImageFolder(train_dir, transform=train_transforms)

我该如何编程地确定数据集中类别或唯一标签的数量？

回答：

使用以下方法：

len(image_datasets['train'].classes)

.classes 返回一个列表。