Replace identity by rotation matrix in spatialtf test

上级 9f489a6c
...@@ -2446,10 +2446,10 @@ def test_dnn_spatialtf(): ...@@ -2446,10 +2446,10 @@ def test_dnn_spatialtf():
downsample_factor = 2 downsample_factor = 2
# Transformation matrix # Transformation matrix
rotation = [[1, 0, 0], theta = [[-1, 0, 0],
[0, 1, 0]] [0, -1, 0]]
transform = np.asarray(img_dims[0] * [rotation], dtype=theano.config.floatX) transform = np.asarray(img_dims[0] * [theta], dtype=theano.config.floatX)
gpu_transform = gpuarray_shared_constructor(transform) gpu_transform = gpuarray_shared_constructor(transform)
st_dnn = dnn.dnn_spatialtf(gpu_img, gpu_transform, downsample_factor) st_dnn = dnn.dnn_spatialtf(gpu_img, gpu_transform, downsample_factor)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论