Как использовать Padding в conv2d слое определенного размера

Question

Мой размер изображения: 256 * 256

Conv2d Kernal Размер: 4*4 и шагает в 2*2.

Выход будет 127*127. Я хочу перейти к Max Pool для этого, я хочу применить отступы, чтобы сделать его 128*128, чтобы пул работал хорошо, и вывод пула будет использоваться в других слоях.

Как я могу применить отступы для этого конв.

conv1 = tf.layers.conv2d(x, 32, (4,4),strides=(2,2), activation=tf.nn.relu)

Answer 1

tf.layers.conv2d имеет параметр padding, который вы можете использовать для этого. По умолчанию "valid", что означает, что заполнение не выполняется, поэтому каждая свертка будет немного сокращать ввод. Вы можете передать padding="same" вместо этого. Это будет применять заполнение таким образом, чтобы вывод свертки был равен по размеру вводу. Это перед шагами, поэтому использование шага 2 все равно приведет к уменьшению с коэффициентом 2. В вашем примере использование padding="same" должно привести к тому, что вывод свертки будет иметь размер 128x128.