Как реализовать дополнения деконволюции кофе в тензор потока? - PullRequest
Новая
       26

Как реализовать дополнения деконволюции кофе в тензор потока?

0 голосов
/ 21 февраля 2019

Этот слой имеет пэд по 4 пикселя с каждой стороны, но в тензорном потоке он просто имеет режим заполнения "SAME" и "VALID".Как реализовать этот слой в тензорном потоке?

layer {
  name: "conv3"
  type: "Deconvolution"
  bottom: "conv26"
  top: "conv3"
  param {
    lr_mult: 0.1
  }
  param {
    lr_mult: 0.1
  }
  convolution_param {
    num_output: 1
    kernel_size: 9
    stride: 3
    pad: 4
    weight_filler {
      type: "gaussian"
      std: 0.001
    }
    bias_filler {
      type: "constant"
      value: 0
    }
  }
}

1 Ответ

0 голосов
/ 21 февраля 2019

Вы можете использовать tf.pad для пользовательского заполнения.Чтобы дополнить тензор с именем my_tensor формой (batch_size, height, width, channels) с четырьмя нулями с обеих сторон для каждого пространственного измерения: 

my_padded_tensor = tf.pad(
    tensor=my_tensor,
    paddings=tf.constant([[0, 0,], [4, 4], [4,4], [0,0]], dtype=tf.int32),
    mode='CONSTANT',
    constant_values=0
)
...