Keras ImageDataGenerator.flow_from_dataframe продолжает загружаться - PullRequest
Новая
       11

Keras ImageDataGenerator.flow_from_dataframe продолжает загружаться

0 голосов
/ 20 ноября 2018

Я работаю над набором данных MURA Стэнфорда.Я пытаюсь загрузить набор данных, используя ImageDataGenerator Keras.Данные находятся в следующей иерархии:

The directory hierarchy

Папка study1_positive содержит изображения.

ImageDataGenerator.flow_from_directory использовать нельзяс этой структурой папок, поэтому я попытался использовать метод flow_from_dataframe.

Однако при запуске код продолжает выполняться и не останавливается.

Ниже приведен формат кадра данных Pandas, который я передаю методу flow_from_directory:

The DataFrame passed to flow_from_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)

1 Ответ

0 голосов
/ 22 ноября 2018

Нашел решение здесь .Очевидно, текущий 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(...)
...