我试图用以下方法训练我的暹罗网络fit_generator()
,我从这个答案中了解到:Keras:如何将 fit_generator 与多个输入一起使用 https://stackoverflow.com/questions/49404993/keras-how-to-use-fit-generator-with-multiple-inputs最好的方法是创建自己的生成器来生成多个数据点,我的问题是我用flow_from_directory()
功能,我不知道这是否可能。
这是我尝试重新调整生成器来解决我的问题:
from keras.models import load_model
from keras import optimizers
from keras.preprocessing.image import ImageDataGenerator
import numpy as np
model = load_model("siamese_model.h5")
train_datagen = ImageDataGenerator(rescale = 1./255)
def generator():
t1 = train_datagen.flow_from_directory(base_dir,target_size = (150, 150), batch_size = 20, class_mode = 'categorical',shuffle = True)
t2 = train_datagen.flow_from_directory(base_dir,target_size = (150, 150), batch_size = 20, class_mode = 'categorical', shuffle = True)
while True:
d1,y = t1.next()
d2 = t2.next()
yield ([d1[0], d2[0]],y)
model.compile(loss = 'categorical_crossentropy',optimizer= optimizers.RMSprop(lr=2e-5),metrics=['acc'])
history = model.fit_generator(generator(),
steps_per_epoch = 10,
epochs = 5)
我的代码给出了与我尝试在没有自定义生成器的情况下拟合模型时完全相同的错误:
ValueError: Error when checking model input: the list of Numpy arrays that you are passing to your model is not the size the model expected. Expected to see 2 array(s), but instead got the following list of 1 arrays: [array([[[[0.14509805, 0.15686275, 0.16862746],
[0.14509805, 0.15686275, 0.16862746],
[0.14509805, 0.15686275, 0.16862746],
...,
[0.14117648, 0.15294118, 0.16862746...
我究竟做错了什么?