ImageDataGenerator 类是用来做数据增强的,通过实时数据增强生成张量图像数据批次。数据将不断循环(按批次)。
注意实时,在玩其.flow_from_directory(directory) 方法时,我发现如果directory里面(就是生成器)有5张图片用于数据增强,而此时我们设置batch_size=3,那么第一次调用生成器会拿到三张图片(比如0,1,3),第二次调用生成器会拿到剩余的2张(2,4)。
如果生成器有5张图片,而batch_size=10,那么调用一次生成器只会拿到5张图片哦。