@Andrei1penguin1

Почему не начинается процесс обучения keras?

Доброго времени суток, есть задача натренировать модель, ищущую лица на изображении
Данные представляют собой следующее: 10000 тренировочных изображений 178x218x3 и 5000 проверочных, а на выходе должно быть 4 числа: координаты крайних углов прямоугольника, ограничивающего область лица
Вот моя модель:
model = models.Sequential()
model.add(layers.Conv2D(32, (3, 3), activation="relu", input_shape=(200, 200, 3)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation="relu"))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(128, (3, 3), activation="relu"))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(128, (3, 3), activation="relu"))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Flatten())
model.add(layers.Dense(512, activation="relu"))
model.add(layers.Dense(4))
model.compile(optimizer="rmsprop", loss="mse", metrics=["mae"])

Вот обучение:
train_dir = "path/to/faces/1"
val_dir = "path/to/faces/2"
train_gen = ImageDataGenerator(rescale=1./255)
val_gen = ImageDataGenerator(rescale=1./255)
train_generator = train_gen.flow_from_directory(train_dir, target_size=(200, 200), batch_size=50, class_mode="categorical")
validation_generator = val_gen.flow_from_directory(val_dir, target_size=(200, 200), batch_size=50, class_mode="categorical")
model.fit_generator(train_generator, steps_per_epoch=200, epochs=3, validation_data=validation_generator, validation_steps=100)

Но при запуске кода летит следующее и после чего ничего не происходит:

Found 0 images belonging to 0 classes.
Found 0 images belonging to 0 classes.
Epoch 1/3

Подскажите пожалуйста, что я не так делаю?
  • Вопрос задан
  • 124 просмотра
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы