dataset中每个类的图像数，PyTorch

Question

我正在处理一个图像数据集，其中图像被分为10类(CIFAR10数据集)。我正在使用PyTorch。我想知道如何通过循环遍历数据集来确定每个类的图像数量。提前感谢您的回复。

Answer 1

你可以把它做两倍。

• 首先创建一个字典img_dict，其中包含CIFAR10数据集的所有类。将所有值初始化为0。
• 下一步循环数据集，并根据img_dict

中的类键继续递增这些值

dataset = CIFAR10(root='data/', download=True, transform=ToTensor())
dataset_size = len(dataset)
classes = dataset.classes
num_classes = len(dataset.classes)
img_dict = {}
for i in range(num_classes):
    img_dict[classes[i]] = 0

for i in range(dataset_size):
    img, label = dataset[i]
    img_dict[classes[label]] += 1

img_dict

您将得到如下输出：

number of images per class:

​