关于
'SAME'填充,
Convolution文档提供了一些详细的解释(这些注释中的更多详细信息)。特别是在使用
'SAME'填充时,输出形状的定义如下:
# for `tf.layers.conv2d` with `SAME` padding:out_height = ceil(float(in_height) / float(strides[1]))out_width = ceil(float(in_width) / float(strides[2]))
在这种情况下,输出形状仅取决于输入形状和步幅。填充大小是从此处计算出来的,以填充此形状要求(而对于
'VALID'填充,则取决于填充大小的是输出形状)
现在,对于转置卷积…由于此操作是法向卷积的后向对应项(其梯度),这意味着法向卷积的输出形状对应于其对应的经转置运算的输入形状。换句话说,虽然的输出形状
tf.layers.conv2d()除以步幅,但的输出形状
tf.layers.conv2d_transpose()却乘以它:
# for `tf.layers.conv2d_transpose()` with `SAME` padding:out_height = in_height * strides[1]out_width = in_width * strides[2]
但是还是要再次计算出填充大小,以获得该输出形状,而不是相反(
SAME填充)。由于这些值之间的正态关系(即找到的关系)为:
# for `tf.layers.conv2d_transpose()` with given padding:out_height = strides[1] * (in_height - 1) + kernel_size[0] - 2 * padding_heightout_width = strides[2] * (in_width - 1) + kernel_size[1] - 2 * padding_width
重新排列我们得到的方程
padding_height = [strides[1] * (in_height - 1) + kernel_size[0] - out_height] / 2padding_width = [[strides[2] * (in_width - 1) + kernel_size[1] - out_width] / 2
注意: 如果例如
2 * padding_height是一个奇数,则padding_height_top =floor(padding_height); 和padding_height_bottom =ceil(padding_height)(与分别相同padding_width,padding_width_left和padding_width_right)
更换
out_height并
out_width与他们的表情,并使用您的值(第1调换卷积):
padding = [2 * (128 - 1) + 4 - (128 * 2)] / 2 = 1
因此
1,您需要在数据的每一侧添加一个填充,以便获得输出暗淡
out_dim = in_dim * stride = strides * (in_dim- 1) + kernel_size - 2 * padding = 256



