Я довольно новичок в Keras / Tensorflow и пытаюсь использовать Keras для задачи классификации лиц, где каждый человек находится в папке, я использую Pytorch, и есть torchvision.datasets.ImageFolder и dataloader для создания набора данныхс преобразованиями и для создания загрузчика данных с указанием размера пакета и т. д. Есть ли способ сделать то же самое с Keras и Tensorflow, я знаю, что есть API данных, но у меня возникли некоторые проблемы, как сделать то же самое с этим.
Любая помощь приветствуется, спасибо
Редактировать Я пытался следовать точно так же, как загрузка данных с помощью tf.data здесь , но я получил ошибку, сказав, что вход имел размеры(нет, ни один, ни один, 3)