keras flow_from_directory返回0张图片

时间:2019-05-09 13:49:33

标签: python keras image-recognition

当我尝试使用以下代码片段尝试对一批图像进行预测时,我收到一条消息,提示未找到图像。

我的目录结构如下。首先,我有一个名为dogs_vs_cats的主目录,在其中有两个子目录train和test,分别包含相应的图像以及包含此代码的笔记本。

test_datagen = ImageDataGenerator()

test_generator = test_datagen.flow_from_directory(
        directory='test/',
        target_size=(300, 300),
        color_mode="rgb",
        shuffle = False,
        class_mode='binary',
        batch_size=1)

filenames = test_generator.filenames

0 个答案:

没有答案