cls_num_list = [0] * args.num_classes # NUM_CLASSES 是你数据集中的类别数量
for label in dset_train.labels:
cls_num_list[label] += 1
print("Class Counts:", cls_num_list)
在LDL代码读取方式上添加读取每类标签个数的列表
最新推荐文章于 2024-07-25 23:32:05 发布