提交 839ce170 authored 作者: João Victor Risso's avatar João Victor Risso

Remove remark on image set resolution in spatial transformer tests

上级 3f858ac7
...@@ -2435,8 +2435,8 @@ def test_dnn_spatialtf(): ...@@ -2435,8 +2435,8 @@ def test_dnn_spatialtf():
grid = T.concatenate([x_t_flat, y_t_flat, ones], axis=0) grid = T.concatenate([x_t_flat, y_t_flat, ones], axis=0)
return grid return grid
# Generate random set of RGB images with 256x256 resolution (pixel values # Generate random set of RGB images (pixel values in [0, 255]). The set of
# in [0, 255]). The set of images is generated in the expected (NCHW) format # images is generated in the expected (NCHW) format
img_dims = (5, 3, 16, 16) img_dims = (5, 3, 16, 16)
img = np.random.randint(low=0, high=256, size=img_dims).astype(theano.config.floatX) img = np.random.randint(low=0, high=256, size=img_dims).astype(theano.config.floatX)
scale_height = 0.25 scale_height = 0.25
...@@ -2493,8 +2493,8 @@ def test_dnn_spatialtf_grad(): ...@@ -2493,8 +2493,8 @@ def test_dnn_spatialtf_grad():
assert any([isinstance(node.op, dnn.GpuDnnTransformerGradT) assert any([isinstance(node.op, dnn.GpuDnnTransformerGradT)
for node in f_gt.maker.fgraph.apply_nodes]) for node in f_gt.maker.fgraph.apply_nodes])
# Generate random set of RGB images with 256x256 resolution (pixel values # Generate random set of RGB images(pixel values in [0, 255]). The set of
# in [0, 255]). The set of images is generated in the expected (NCHW) format # images is generated in the expected (NCHW) format
input_dims = (5, 3, 16, 16) input_dims = (5, 3, 16, 16)
inputs_val = np.random.randint(low=0, high=256, size=input_dims).astype(theano.config.floatX) inputs_val = np.random.randint(low=0, high=256, size=input_dims).astype(theano.config.floatX)
# Tensor with transformations # Tensor with transformations
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论