diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/python/test_rec_scaling.py | 27 |
1 files changed, 24 insertions, 3 deletions
diff --git a/tests/python/test_rec_scaling.py b/tests/python/test_rec_scaling.py index 621fd8a..ee27efc 100644 --- a/tests/python/test_rec_scaling.py +++ b/tests/python/test_rec_scaling.py @@ -76,7 +76,7 @@ def ProjectionGeometries(type): class TestRecScale(unittest.TestCase): - def single_test(self, geom_type, proj_type, alg, iters): + def single_test(self, geom_type, proj_type, alg, iters, vss, dss): if alg == 'FBP' and 'fanflat' in geom_type: self.skipTest('CPU FBP is parallel-beam only') is3D = (geom_type in ['parallel3d', 'cone']) @@ -88,7 +88,12 @@ class TestRecScale(unittest.TestCase): else: vol = np.zeros((64,64,64),dtype=np.float32) vol[25:35,25:35,25:35] = 1 - proj_id = astra.create_projector(proj_type, pg, vg) + options = {} + if vss > 1: + options["VoxelSuperSampling"] = vss + if dss > 1: + options["DetectorSuperSampling"] = vss + proj_id = astra.create_projector(proj_type, pg, vg, options=options) if not is3D: sino_id, sinogram = astra.create_sino(vol, proj_id) else: @@ -185,6 +190,14 @@ __algs_cone = { } +__combinations_ss = { + 'parallel': [ { 'projector': 'cuda', 'alg': 'SIRT_CUDA', 'iters': 50 } ], + 'fanflat': [ { 'projector': 'cuda', 'alg': 'SIRT_CUDA', 'iters': 50 } ], + 'parallel3d': [ { 'projector': 'cuda3d', 'alg': 'SIRT3D_CUDA', 'iters': 200 } ], + 'cone': [ { 'projector': 'cuda3d', 'alg': 'SIRT3D_CUDA', 'iters': 200 } ] +} + + for k, l in __combinations.items(): for v in l: @@ -199,7 +212,7 @@ for k, l in __combinations.items(): A = __algs for a, i in A.items(): def f(k, v, a, i): - return lambda self: self.single_test(k, v, a, i) + return lambda self: self.single_test(k, v, a, i, 1, 1) setattr(TestRecScale, 'test_' + a + '_' + k + '_' + v, f(k,v,a,i)) for k, l in __combinations_adjoint.items(): @@ -208,6 +221,14 @@ for k, l in __combinations_adjoint.items(): return lambda self: self.single_test_adjoint3D(k, v) setattr(TestRecScale, 'test_adjoint_' + k + '_' + v, g(k,v)) +for k, l in __combinations_ss.items(): + for A in l: + for vss in [1, 2]: + for dss in [1, 2]: + def h(k, v, a, i, vss, dss): + return lambda self: self.single_test(k, v, a, i, vss, dss) + setattr(TestRecScale, 'test_ss_' + a + '_' + k + '_' + v + '_' + str(vss) + '_' + str(dss), h(k, A['projector'], A['alg'], A['iters'], vss, dss)) + if __name__ == '__main__': unittest.main() |