提交 7e41d0c4 authored 作者: Duc Nguyen's avatar Duc Nguyen

fixed shape error of frac bilinear upsample

上级 65b313d5
...@@ -1855,9 +1855,7 @@ def frac_bilinear_upsampling(input, ...@@ -1855,9 +1855,7 @@ def frac_bilinear_upsampling(input,
# upsample the input by passing it as kernel of conv and using filter_dilation # upsample the input by passing it as kernel of conv and using filter_dilation
upsamp = T.nnet.conv2d(pad_kern, concat_mat, border_mode='valid', upsamp = T.nnet.conv2d(pad_kern, concat_mat, border_mode='valid',
filter_dilation=ratio, subsample=subsample) filter_dilation=ratio, subsample=subsample)
return upsamp.reshape((input.shape[0], input.shape[1], upsamp.shape[2], upsamp.shape[3]))
up_img_sh = T.ceil(T.as_tensor([row, col]) * np.array(ratio) / np.array(subsample)).astype('int64')
return upsamp.reshape((input.shape[0], input.shape[1], up_img_sh[0], up_img_sh[1]))
def bilinear_upsampling(input, def bilinear_upsampling(input,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论