Keras ImageDataGenerator.flow_from_dataframe 继续加载答案

【问题标题】：Keras ImageDataGenerator.flow_from_dataframe keeps on loadingKeras ImageDataGenerator.flow_from_dataframe 继续加载
【发布时间】：2019-04-22 16:27:09
【问题描述】：

我正在研究斯坦福大学的MURA dataset。我正在尝试使用 Keras 的 ImageDataGenerator 加载数据集。数据位于以下层次结构中：

study1_positive 文件夹包含图像。

ImageDataGenerator.flow_from_directory 不能用于此文件夹结构，因此我尝试使用flow_from_dataframe 方法。

但是，当运行时，代码会继续执行并且不会停止。

以下是我传递给flow_from_directory 方法的 Pandas DataFrame 的格式：

我还尝试将标签分别更改为“异常”和“正常”来代替 1 和 0。

编辑：

我已经粘贴了下面的代码：

train_imggen = ImageDataGenerator(rescale=1./255, rotation_range=30,
                              horizontal_flip=True)

train_loader = train_imggen.flow_from_dataframe(traindf, '.', shuffle=True,
                                            x_col='path', y_col='label',
                                            color_mode='grayscale',
                                            target_size=(320,320), 
                                            class_mode='binary', 
                                            batch_size=8)

【问题讨论】：

没有代码就没有办法帮助你。请添加。
@MatiasValdenegro 道歉！我现在已经添加了生成器代码。

标签： python python-3.x keras dataset

【解决方案1】：

找到解决方案here。显然，当前的flow_from_dataframe 不处理相对路径。

克隆我的“fix_found_0_images”分支。 git clone -b fix_found_0_images_bug https://github.com/smurak/keras-preprocessing.git

将“keras_preprocessing”子目录移至您的工作目录。

导入它。

import keras
from keras_preprocessing import image
...
train_imggen = image.ImageDataGenerator(...)

【讨论】：