具有不同滤波器尺寸的张量流卷积

时间:2017-07-13 00:55:21

标签: python tensorflow conv-neural-network convolution

我想用不同大小的过滤器对我的数据源进行卷积,并想知道如何使用Tensorflow实现以下设置 enter image description here

换句话说,我希望有两个平行的卷积并将它们连接到展平层,就在进入完全连接的层之前,但我不确定连接。

任何代码片段或方法上的来源都会给您带来极大的帮助!

1 个答案:

答案 0 :(得分:1)

假设批量大小100和大小为28x28x1的图像数据。

import tensorflow as tf

inp = tf.placeholder(tf.float32, shape=[100, 28, 28, 1])
left_branch = tf.layers.conv2d(input=inp, filters=N, kernel_size=[L, M])
right_branch = tf.layers.conv2d(input=inp, filters=P, kernel_size=[R, S])

left_reshape = tf.reshape(left_branch, [100, num_outputs_in_left_branch])
right_reshape = tf.reshape(right_branch, [100, num_outputs_in_right_branch])

combined_branch = tf.concat([left_reshape, right_reshape], axis=1)
combined_branch = tf.layers.dense(combined_branch, num_units_in_dense)