【发布时间】:2020-05-10 05:14:14
【问题描述】:
tf.keras.Concatenate() 方法遇到了一个奇怪的问题。我试图执行组卷积。这是代码。
# C2: (None, 27, 27, 96) -> (None, 27, 27, 256).
# Split (None, 27, 27, 96) into x2 (None, 27, 27, 48)
pool1_1 = Lambda(lambda x: x[:, :, :, :48])(pool1)
pool1_2 = Lambda(lambda x: x[:, :, :, 48:])(pool1)
#####################
# Grouped convolution.
#####################
conv2_1 = Conv2D(filters=128,
kernel_size=(5,5),
activation='relu',
padding='same',
name='conv2_1')(pool1_1)
conv2_2 = Conv2D(filters=128,
kernel_size=(5,5),
activation='relu',
padding='same',
name='conv2_2')(pool1_2)
conv2 = Concatenate(name='conv2', axis=-1)([conv2_1, conv2_2])
这里是output。
如您所见,在连接之后,结果层的参数为 0。我希望它有 153728 * 2 个参数。这是为什么呢?
【问题讨论】:
标签: keras deep-learning computer-vision conv-neural-network tensorflow2.0