提交 3f858ac7 authored 作者: João Victor Risso's avatar João Victor Risso

Use tensor.ceil in out_dims setup in dnn_spatialtf function

上级 4d85a8c3
......@@ -3015,8 +3015,8 @@ def dnn_spatialtf(img, theta, scale_width=1, scale_height=1, precision=theano.co
Also, the only grid sampler method available is the bilinear interpolation.
"""
out_dims = (img.shape[0], img.shape[1],
img.shape[2] * scale_height,
img.shape[3] * scale_width)
theano.tensor.ceil(img.shape[2] * scale_height),
theano.tensor.ceil(img.shape[3] * scale_width))
out_dims = tuple([as_scalar(v).astype('int64') for v in out_dims])
# Create spatial transformer descriptor
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论