Предупреждение о наборе небольших изображений Tensorflow 2.0

Question

Я изучаю классификацию изображений с помощью TensorFlow 2.0, но при ее реализации я получаю следующее предупреждение:

тензор потока: в вашем вводе закончились данные; прерывающее обучение. Убедитесь, что ваш набор данных или генератор может генерировать как минимум steps_per_epoch * epochs пакетов (в данном случае 11280 пакетов). Вам может понадобиться использовать функцию repeat () при построении набора данных.

Вот размеры набора данных и данные модели:

total training mango images : 752
total validation mango images : 288

history = model.fit_generator(train_generator,
                              validation_data=validation_generator,
                              steps_per_epoch=752,
                              epochs=15,
                              validation_steps=288)

Я также использовал методы увеличения изображения, чтобы сделать модель эффективной. но, тем не менее, предупреждение прерывает обучение, и модель остается на подходящем.

Вот ссылка на мою полную модель в github

любезно помогите мне!

Заранее спасибо

Answer 1

Учитывая, что вы получили ошибку на 24-м шаге и что batch_size равно 32, я полагаю, проблема в том, что ваша 24-я партия имеет только 16 изображений. Вы можете удалить эти изображения, чтобы ваш набор данных выровнялся с размером пакета 32 или изменить batch_size на 16.

Кстати, я подозреваю, что вы используете steps_per_epoch и validation_steps неправильно - «шаги» означает количество партий, а не количество образцов. FYI