datagen.flow_from_directory()怎么判断是调用那个文件夹的图片
时间: 2024-05-29 08:14:23 浏览: 138
tensorflow下的图片标准化函数per_image_standardization用法
`datagen.flow_from_directory()` 方法会根据指定的文件夹路径,自动扫描该文件夹下的子文件夹和文件,来获取训练数据和标签。其中,该方法的 `directory` 参数指定了包含子文件夹的文件夹路径,`target_size` 参数指定了训练数据的大小,`class_mode` 参数指定了分类模式(例如二分类或多分类),`batch_size` 参数指定了每个批次的大小,等等。
因此,该方法会根据文件夹中的子文件夹和文件名,自动判断哪些图片属于哪个类别,并建立对应的标签。例如,假设有一个文件夹 `data`,其中包含两个子文件夹 `cat` 和 `dog`,分别存放了猫和狗的图片。那么,调用 `datagen.flow_from_directory(directory='data', target_size=(224, 224), class_mode='binary', batch_size=32)` 方法时,会返回一个迭代器,每次迭代会返回一个包含 32 张图片和对应标签的批次数据。在这个例子中,模型会自动将 `cat` 文件夹中的图片标记为 0,将 `dog` 文件夹中的图片标记为 1,从而训练一个二分类的模型。
阅读全文