From 1423e0b9bbb12d6be588e09dd3f73ae2e3646c56 Mon Sep 17 00:00:00 2001 From: "Georg Schramm (Windows)" Date: Mon, 23 Sep 2024 16:59:51 +0200 Subject: [PATCH] add new data set in test script --- test_petric.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/test_petric.py b/test_petric.py index e91419e..02940ec 100644 --- a/test_petric.py +++ b/test_petric.py @@ -345,6 +345,9 @@ def test_petric(ds: int, num_iter: int, suffix: str = "", **kwargs): elif ds == 4: srcdir = SRCDIR / "Mediso_NEMA_IQ" outdir = OUTDIR / "Mediso_NEMA_IQ" / sdir_name + elif ds == 5: + srcdir = SRCDIR / "Siemens_mMR_NEMA_IQ_lowcounts" + outdir = OUTDIR / "mMR_NEMA_lowcounts" / sdir_name else: raise ValueError(f"Unknown data set {ds}") @@ -421,14 +424,9 @@ def test_petric(ds: int, num_iter: int, suffix: str = "", **kwargs): ) else: for ns in [25]: - for i in range(5): - for fwhm in [3.0, 1.0, 7.0]: - phf = 0.75 - test_petric( - ds=i, - num_iter=200, - approx_num_subsets=ns, - suffix=f"ns_{ns}_phf_{phf}_fwhm_{fwhm}", - precond_hessian_factor=0.75, - precond_filter_fwhm_mm=fwhm, - ) + for i in [5]: + test_petric( + ds=i, + num_iter=200, + approx_num_subsets=ns, + )