keras从数据生成器ImageDataGenerator中的flow_from_directory中获取加载的文件名和相应的类别

最新推荐文章于 2024-08-08 02:58:38 发布

wf592523813

最新推荐文章于 2024-08-08 02:58:38 发布

阅读量4.4k

点赞数 1

分类专栏： keras 文章标签： flow_from_directory

本文链接：https://blog.csdn.net/wf592523813/article/details/95599345

版权

keras 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

最近在做图像分类的任务，使用了densenet进行fine-tune，因为图片数据比较多且占内存因此数据的加载使用ImageDataGenerator生成器，使用flow_from_directory从文件夹中获取各个类别的数据。

因为在测试的时候，需要知道哪个数据被判错了，要找到对应的文件名。

test_datagen = ImageDataGenerator(rescale=1./255)
val_generator = test_datagen.flow_from_directory( test_dir,
                                                  target_size=(img_size, img_size),
                                                  batch_size=32,
                                                  shuffle=False)
print(val_generator.class_indices)   # 输出对应的标签文件夹
print(val_generator.filenames)  # 按顺序输出文件的名字