提交 cfdf4841 authored 作者: Sina Honari's avatar Sina Honari

adding the dimensionality of images by one

上级 1f615fcf
...@@ -217,7 +217,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -217,7 +217,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
maxpoolshps = ((1, 1), (3, 3), (5, 3)) maxpoolshps = ((1, 1), (3, 3), (5, 3))
stridesizes = ((1, 1), (3, 3), (5, 7)) stridesizes = ((1, 1), (3, 3), (5, 7))
imval = rng.rand(1, 1, 16, 16) imval = rng.rand(1, 2, 16, 16)
for maxpoolshp in maxpoolshps: for maxpoolshp in maxpoolshps:
for ignore_border in [True, False]: for ignore_border in [True, False]:
...@@ -240,7 +240,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -240,7 +240,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
for indx in numpy.arange(len(maxpoolshps)): for indx in numpy.arange(len(maxpoolshps)):
imvsize = imvsizs[indx] imvsize = imvsizs[indx]
imval = rng.rand(1, 1, imvsize[0], imvsize[1]) imval = rng.rand(1, 2, imvsize[0], imvsize[1])
stride = stridesizes[indx] stride = stridesizes[indx]
maxpoolshp = maxpoolshps[indx] maxpoolshp = maxpoolshps[indx]
for ignore_border in [True, False]: for ignore_border in [True, False]:
...@@ -280,7 +280,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -280,7 +280,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
maxpoolshps = ((1, 1), (3, 3), (5, 3)) maxpoolshps = ((1, 1), (3, 3), (5, 3))
stridesizes = ((1, 1), (3, 3), (5, 7)) stridesizes = ((1, 1), (3, 3), (5, 7))
imval = rng.rand(1, 1, 16, 16) imval = rng.rand(1, 2, 16, 16)
for maxpoolshp in maxpoolshps: for maxpoolshp in maxpoolshps:
for ignore_border in [True, False]: for ignore_border in [True, False]:
...@@ -313,7 +313,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -313,7 +313,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
for indx in numpy.arange(len(maxpoolshps)): for indx in numpy.arange(len(maxpoolshps)):
imvsize = imvsizs[indx] imvsize = imvsizs[indx]
imval = rng.rand(1, 1, imvsize[0], imvsize[1]) imval = rng.rand(1, 2, imvsize[0], imvsize[1])
stride = stridesizes[indx] stride = stridesizes[indx]
maxpoolshp = maxpoolshps[indx] maxpoolshp = maxpoolshps[indx]
for ignore_border in [True, False]: for ignore_border in [True, False]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论