title: Creating image dataset from folder in Keras
body: <p>I am pretty new to Keras/Tensorflow and I am trying to use Keras for a face classification task where each person is in a folder, I have been using Pytorch and there is torchvision.datasets.ImageFolder and dataloader to create a dataset with transformations and to create a data loader specifying batchsize etc. Is there a way to do the same with Keras and Tensorflow, I know there is Data API, but I have been having some trouble how to do the same with using that.</p>
  
  <p>Any help is appreciated thank you</p>
  
  <p>Edit I tried to follow the exact same thing as <a href="https://www.tensorflow.org/tutorials/load_data/images" rel="nofollow noreferrer">loading data guide with tf.data here</a> but I got an error saying the input had dimensions (none,none,none,3)</p>