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

【问题标题】：Keras flow_from_directory won't load images from driveKeras flow_from_directory 不会从驱动器加载图像
【发布时间】：2020-04-27 20:12:52
【问题描述】：

我正在尝试使用 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.

如果有人可以提供帮助将不胜感激！ :)

【问题讨论】：

Please don't post your code as images..
哦，好吧，对不起，我会重新格式化我的帖子
您是否将 gdrive 安装为“从 google.colab 导入驱动器 drive.mount('/content/gdrive')”？您在flow_from_directory 中使用了正确的路径吗？
@VishnuvardhanJanapati 我安装在 '/content/drive' 并且我在 flow_from_directory 中使用相同的路径
您的图片格式是什么？ PNG？ JPEG？

标签： python keras google-colaboratory

【解决方案1】：

好的，我刚刚发现了错误，我在使用 flow_from_directory() 时将“分类”用于属性类而不是用于 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')

【讨论】：