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