我正在从正确形式的input_shape寻求帮助,
如果我有类似以下的代码,
G = ImageGenerator(batch_size,args,****kwargs)
其中G是一个定制的data_generator,它指定批处理大小,然后将G生成的数据拟合到:
m.fit_generator(G ,#some other parameters)
我知道在模型的第一层,有必要给出input_shape
。
我的问题是:
在这种情况下,已在生成器中声明了批量大小G
,
构建第一层时,我是否必须告诉Keras(传递参数):
batch_size=32,input_shape=(256,256)
或者只是
input_shape=(256,256)
谢谢!
答案 0 :(得分:0)
根据Matias Valdenegro的说法:批量大小不是任何图层的参数,只需输入形状即可。