需要在Keras中指定输入形状

时间:2019-03-08 09:11:21

标签: deep-learning batch-processing shapes keras-layer

我正在从正确形式的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)

谢谢!

1 个答案:

答案 0 :(得分:0)

根据Matias Valdenegro的说法:批量大小不是任何图层的参数,只需输入形状即可。