gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-30 09:43:49 30 4
gpt4 key购买 nike

如果我有一个像这样的数据集:

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

如何以编程方式确定数据集中的类或唯一标签的数量?

最佳答案

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

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

关于python - pytorch:获取给定 ImageFolder 数据集的类数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55235594/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com