train_generator = train_datagen.flow_from_directory( train_dir, target_size= array_image.shape[:2], batch_size = batch_size, color_mode= "rgb", class_mode= "categorical")
时间: 2023-03-28 10:02:55 浏览: 175
train_data-数据集
5星 · 资源好评率100%
这是一个关于使用 Keras 中的 ImageDataGenerator 生成训练数据的问题,我可以回答。这段代码是用于从指定目录中读取图像数据,并将其转换为指定大小和颜色模式的张量,同时生成对应的标签。这些数据和标签将被用于训练深度学习模型。
阅读全文