Набор данных TensorFlow из списка изображений в модели керас

Оказывается, проблема заключалась в использовании модели Keras. Пример из учебника TF основан на построении модели Keras с использованием модуля tf.keras (все слои, модели и т. Д. Взяты из tf.keras). Хотя модель, которую я использовал (DenseNet), основана на модуле чистого keras, то есть все слои были получены из модуля keras, а не из tf.keras. Это приводит к тому, что tf.data.Dataset проверяется на наличие ndim в методе подгонки модели keras. Как только я настроил свою DenseNet на использование слоев tf.keras, все снова стало работать.