gpt4 book ai didi

python - Keras - flow_from_directory : How to get number of samples in each category?

转载 作者:行者123 更新时间:2023-12-02 02:26:06 25 4
gpt4 key购买 nike

Keras 有没有办法在使用 ImageDataGenerator 时获取每个类别中的图像数量?假设我有一个包含 3 个子文件夹(类)的训练文件夹,每个子文件夹都包含一定数量的图像。

imagedatagen = ImageDataGenerator(...)    

imageflow = imagedatagen.flow_from_directory(
source_dir,
class_mode='categorical', ...)

我可以使用以下命令获取类映射或所有样本的数量:

class_dict = imageflow.class_indices # returns ie. {'aligator': 0, 'cat': 1, 'dog': 2}
sample_count = imageflow.samples # returns sum of all samples, not divided into categories

但是我如何获得信息,类别 0 包含 1000 张图片,类别 1 包含 1500 张图片,类别 2 包含其余所有图片?我知道我可以在 os 模块的帮助下迭代所有文件,但想知道是否还有其他方法,但我找不到它。

最佳答案

据我所知,没有内置的方法可以实现这一点。

您可以尝试使用 collections模块。

from collections import Counter

counter = Counter(imageflow.classes)

print(counter.items()) # dict_items([(0, 1648), (1, 3614)])

关于python - Keras - flow_from_directory : How to get number of samples in each category?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65632501/

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