import time
from keras.preprocessing.image import ImageDataGenerator
import numpy as np
imgs=np.random.randint(0,10,size=(7,100,100,3))
datagen = ImageDataGenerator(
featurewise_center=True,
featurewise_std_normalization=True,
rotation_range=20,
width_shift_range=0.2,
height_shift_range=0.2,
horizontal_flip=True)
f=datagen.flow(imgs,[0,1,2,3,4,5,6],batch_size=3)
# print(f.next()[1])
# time.sleep(2)
# print(f.next()[1])
# time.sleep(2)
# print(f.next()[1])
for index,(x,y) in enumerate(f):
if index==10:
break
time.sleep(1)
print(x.shape,y)
输出结果
(3, 100, 100, 3) [5 0 6]
(3, 100, 100, 3) [3 2 1]
(1, 100, 100, 3) [4]
(3, 100, 100, 3) [6 0 4]
(3, 100, 100, 3) [1 2 3]
(1, 100, 100, 3) [5]
(3, 100, 100, 3) [6 1 0]
(3, 100, 100, 3) [4 5 2]
(1, 100, 100, 3) [3]
。。。。。。
打乱,批处理之后,不断循环