Keras ImageDataGenerator.flow_from_dataframe继续加载

Question

我正在研究斯坦福大学的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)

Answer 1

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

  

  
1. 克隆我的“ fix_found_0_images”分支。 git clone -b fix_found_0_images_bug https://github.com/smurak/keras-preprocessing.git
  
2. 将“ keras_preprocessing”子目录移动到您的工作目录中。
  
3. 导入。
  

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