keras ImageDataGenerator.flow()和ImageDataGenerator.flow_from_directory()有什么区别?

jtw3ybtb  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(138)
datagen = ImageDataGenerator(rotation_range = 20, zoom_range = 15, width_shift_range=0.2, 
    height_shift_range=0.2, shear_range=0.15, horizontal_flip=True, 
    vertical_flip=True, fill_mode='nearest')

datagen.flow()datagen.flow_from_directory的区别是什么?

8fsztsew

8fsztsew1#

datagen.flow():此方法期望输入数据作为Numpy数组提供。当您已经将整个数据集作为Numpy数组加载到内存中时,通常会使用此方法。将图像的Numpy数组及其对应的标签传递给flow()方法,以生成批量的增强数据。
datagen.flow_from_directory():此方法旨在处理以目录结构组织的图像数据,其中每个子目录表示一个类并包含属于该类的图像。该方法自动从目录结构推断类标签。您只需要指定包含图像的目录,该方法将生成具有适当标签的批量增强数据。

相关问题