Если у меня есть такой набор данных:
image_datasets['train'] = datasets.ImageFolder(train_dir, transform=train_transforms)
Как программно определить количество классов или уникальных меток в наборе данных?
Использование:
len(image_datasets['train'].classes)
.classes возвращает список.
.classes