Friday, May 7, 2021

Dataset from ImageDataGenerator 
I have created a Image Data Generator. Now can I convert it to a keras dataset ? 

Yes. You can use tf.data.Dataset.from_generator:


 tf.data.Dataset.from_generator(lambda: ImageDataGenerator().flow_from_directory('folder_path'),(tf.float32, tf.float32))

No comments:

Post a Comment

useful links

  https://github.com/huggingface/cookbook/tree/main/notebooks/en  hugging face ai cookbooks  https://colab.research.google.com/github/huggin...