ValueError: `generator` вернула элемент формы (2, 200, 200, 3), где ожидался элемент формы (4, 200, 200, 3)

Question

Я создаю свою функцию генератора с batch_size = 4, как показано ниже:

tf.data.Dataset.from_generator(img_gen, output_types=(tf.float32, tf.float32), output_shapes = ([4,200,200,3],[4,2]))

После запуска нахожу эту ошибку

ValueError: `generator` yielded an element of shape (2, 200, 200, 3) where an element of shape (4, 200, 200, 3) was expected.

Я не могу понять почему?

Answer 1

Вероятная причина в том, что количество примеров данных не делится на размер пакета. Либо расположите ваши данные так, чтобы они были кратны размеру пакета, либо, лучше, пусть последующий код работает с любым размером пакета.