谷歌云机器学习引擎(Google Cloud Machine Learning Engine)是谷歌云平台上的一项托管式机器学习服务,它提供了一个强大的基础设施,用于训练和部署机器学习模型。在谷歌云机器学习引擎上运行时使用Keras的flow_from_directory
函数,可以方便地从目录中加载大量的图像数据进行模型训练。
Keras是一个流行的深度学习框架,它提供了高级的API,使得构建和训练深度神经网络变得简单而直观。flow_from_directory
是Keras中一个非常有用的函数,它可以从指定的目录中自动加载图像数据,并进行数据预处理、批量生成训练样本。
使用flow_from_directory
函数时,需要将图像数据按照类别分别放置在不同的子目录中。该函数会自动遍历指定目录及其子目录,将图像数据加载到内存中,并根据目录结构自动为每个类别分配一个标签。这样,我们就可以方便地从目录中加载大量的图像数据,并将其用于模型的训练。
flow_from_directory
函数的一些常用参数包括:
directory
:指定包含图像数据的目录路径。target_size
:指定图像的目标尺寸,通常是一个元组,如(224, 224)
。batch_size
:指定每个批次的样本数量。class_mode
:指定类别的标签类型,常用的取值包括"categorical"
(多分类问题)和"binary"
(二分类问题)。shuffle
:指定是否在每个epoch之前打乱数据。