FCN的tf.nn.conv2d_transpose output_shape

Question

I want to implement deconvolution layer in tensorflow for FCN model, I used tf.nn.conv2d_transpose for each of 5 conv outputs, what I need is that the output shape of each of the 5 deconv to be the same as the input image shape. So I set

deconv_shape = tf.shape(input)
tf.nn.conv2d_transpose(value=deconv5_1,
                       filter=[32, 32, 1, 1],
                       output_shape=deconv_shape,
                       strides=16,
                       padding="same",
                       name="deconv5_2")

Am I doing it right?

Answer 1

I think your implementation isn't correct, here's the few step to get it right.

in_channels = input.shape[-1]
# here set the output_height, width as [stride*input_height, stride*input_width]]
output_shape = [batch_size, output_height, output_width, out_channels]
filter_size =2 # for example 
stride = 2 # for example if you want 2x scale of input height, width
shape = [filter_size, filter_size, out_channels, in_channels]
w = tf.get_variable(
        name='W',
        shape=shape,
        initializer=w_init,
        regularizer=w_regularizer,
        trainable=trainable
)

output = tf.nn.conv2d_transpose(
        input, w, output_shape=output_shape, strides=[1, stride, stride, 1])