提交 7949f5e4 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix output shape in 1D upsampling

上级 2ef606f6
......@@ -522,6 +522,7 @@ def bilinear_upsampling(input,
# first and last col
concat_mat = T.concatenate((concat_mat[:, :, :, :1], concat_mat,
concat_mat[:, :, :, -1:]), axis=3)
concat_col = col + 2
pad = 2 * ratio - (ratio - 1) // 2 - 1
......@@ -533,7 +534,8 @@ def bilinear_upsampling(input,
np.newaxis, :,
np.newaxis],
input_shape=(up_bs, 1,
row * ratio, col),
row * ratio,
concat_col),
filter_shape=(1, 1, None, 1),
border_mode=(pad, 0),
subsample=(ratio, 1),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论