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

2024-03-28 10:52:46 发布

您现在位置:Python中文网/ 问答频道 /正文

如果我有如下数据集:

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

如何通过编程确定数据集中类或唯一标签的数量


Tags: 数据image数量编程dirtransformtrain标签
2条回答

使用:

len(image_datasets['train'].classes)

.classes返回一个列表

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

import torch n_classes = len(torch.unique(Your_Target_Vector))

相关问题 更多 >