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

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