Tensorflow 从Keras中的文件夹创建图像数据集

Tensorflow 从Keras中的文件夹创建图像数据集,tensorflow,keras,Tensorflow,Keras,我是Keras/Tensorflow的新手,我正在尝试使用Keras进行人脸分类任务,每个人都在一个文件夹中,我一直在使用Pytorch,有torchvision.datasets.ImageFolder和dataloader来创建带有转换的数据集,并创建指定batchsize等的数据加载器。有没有办法对Keras和Tensorflow进行同样的操作,我知道有数据API,但我在使用它时遇到了一些问题 谢谢你的帮助 编辑我试图遵循与的完全相同的操作,但我得到一个错误,说输入有维度(无,无,无,3)

我是Keras/Tensorflow的新手,我正在尝试使用Keras进行人脸分类任务,每个人都在一个文件夹中,我一直在使用Pytorch,有torchvision.datasets.ImageFolder和dataloader来创建带有转换的数据集,并创建指定batchsize等的数据加载器。有没有办法对Keras和Tensorflow进行同样的操作,我知道有数据API,但我在使用它时遇到了一些问题

谢谢你的帮助


编辑我试图遵循与的完全相同的操作,但我得到一个错误,说输入有维度(无,无,无,3)

看看这里的示例:

我相信你会发现这个有用吗?如果没有,请告诉我