当我尝试训练模型时,以下代码不断出现以下错误: TypeError: fit_generator() missing 1 required positional argument: 'generator'
。我怎么也找不出是什么导致了这个错误。x_train 是一个形状为 (400, 256, 256, 3) 的RGB图像,而对于 y_train,我有10个输出类,使其形状为 (400, 256, 256, 10)。这里出了什么问题?
如果需要,可以通过以下链接下载数据:https://www49.zippyshare.com/v/5pR3GPv3/file.html
回答:
你的代码中有几个错误,但考虑到你的错误:
TypeError: fit_generator() missing 1 required positional argument: ‘generator’
这是因为 fit_generator 调用了 XYaugmentGenerator,但内部没有调用增强生成器。
gen.flow(...
不会工作,因为 gen 没有被声明。你应该要么将 image_datagen 重命名为 gen,如下所示:
gen = ImageDataGenerator(**data_gen_args)
或者,将 gen 替换为 image_datagen
genX1 = image_datagen.flow(X1, y, batch_size=batch_size, seed=seed)genX2 = image_datagen.flow(y, X1, batch_size=batch_size, seed=seed)