我对Keras / Tensorflow还是很陌生，我尝试使用Keras进行人脸分类任务，其中每个人都在一个文件夹中，我一直在使用Pytorch，并有torchvision.datasets.ImageFolder和dataloader创建数据集通过转换并创建一个指定批处理大小的数据加载器等。是否有办法对Keras和Tensorflow进行同样的处理，我知道有Data API，但是我在使用该方法做同样的事情时遇到了一些麻烦。 >
感谢您的任何帮助，谢谢
编辑，我尝试遵循与loading data guide with tf.data here完全相同的方法，但是出现错误，指出输入的尺寸为（none，none，none，3）