当我尝试使用以下代码片段来尝试对一批图像进行预测时,我收到一条消息说没有找到图像:
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
我的目录结构如下。我首先有一个名为 dogs_vs_cats 的主目录,其中有两个子目录 train 和 test 包含各自的图像以及包含此代码的笔记本。