图像深度学习的预测
#对应的训练集label
train_datagen.flow_from_directory
#预测
#需要在这之前先读取预测集label
test_datagen = ImageDataGenerator(rescale=1. / 255)
test_generator = test_datagen.flow_from_directory(
test_dir,
target_size=(128, 128),
batch_size=20,
class_mode='categorical',
shuffle=False)
#然后预测
model.predict_generator