Keras flow_from_directory不会从驱动器加载图像

问题描述 投票:0回答:1

我正在尝试使用keras flow_from_directorY()加载图像以训练CNN,但是似乎很难找到图像。我正在google colab上运行代码

!unzip "/content/drive/My Drive/fruits.zip" -d "/content/drive/My Drive"

我使用此命令解压缩数据集,然后,如果我尝试显示每个文件夹有多少个图像,则似乎可以识别图像

for folder in os.listdir(os.path.join(path,'training')):
folder_count = len(os.listdir(os.path.join(path,'training',folder)))
image_count.append(folder_count)
class_names.append(folder)
print('{:20s}'.format(folder), end='')
print(folder_count)
print('-'*24)
print('Number of Classes:', len(class_names))
print('Average number of images per Class: ', 
np.array(image_count).mean())

但是,如果我尝试将flow_from_directory与这些路径一起使用,我发现它找到了0张图像

train_batches = train_datagenerator.flow_from_directory('/content/drive/My 
Drive/fruits-360_dataset/data/training', target_size=(224, 224), 
color_mode="rgb", classes="categorical" ,  batch_size=32, subset='training')
val_batches = train_datagenerator.flow_from_directory('/content/drive/My 
Drive/fruits-360_dataset/data/training', target_size=(224, 224), 
color_mode="rgb", classes="categorical" ,  batch_size=32, 
subset='validation')
test_batches = test_datagenerator.flow_from_directory('/content/drive/My 
Drive/fruits-360_dataset/data/test', target_size=(224, 224), 
color_mode="rgb", classes="categorical" ,  batch_size=32, shuffle=False)

Found 0 images belonging to 11 classes.
Found 0 images belonging to 11 classes.
Found 0 images belonging to 11 classes.

如果有人可以提供帮助,我们将不胜感激! :)

python keras google-colaboratory
1个回答
0
投票

[确定,我刚刚发现了错误,我在使用flow_from_directory()时对属性类使用了'categorical'而不是class_mode

train_batches = train_datagenerator.flow_from_directory('/content/drive/My 
Drive/fruits-360_dataset/data/training', target_size=(224, 224), 
color_mode="rgb", class_mode="categorical" ,  batch_size=32, subset='training')
© www.soinside.com 2019 - 2024. All rights reserved.