python - 如何修复 Keras 中的批量大小？

Question

基本上，我想编写一个损失函数来计算比较标签和批处理输出的分数。为此，我需要修复批量大小。

我之前是在 Tensorflow 中完成的，我可以在占位符函数中设置批量大小。现在，我需要在提供给我的 Keras 代码中使用类似的机制。我不知道这里该怎么做。

conv1 = (Conv2D(32, (3,3), padding='same', kernel_regularizer=regularizers.l2(weight_decay), input_shape=x_train.shape[1:], activation='elu'))(input_img)
print(conv1.shape)

打印语句的输出显然是[?, 32, 32, 3]。我该如何做到这一点，比如[64, 32, 32, 3]？

Answer 1

使用keras.layers.Input层指定批量大小:

from keras.layers import Conv2D, Input
from keras import regularizers

x = Input(shape=(32, 32, 3), batch_size=64)
conv1 = Conv2D(filters=32,
               kernel_size=(3,3),
               padding='same',
               kernel_regularizer=regularizers.l2(1.),
               input_shape=(32, 32, 3),
               activation='elu')(x)
print(conv1.shape) # (64, 32, 32, 32)