def test_shepp_logan_SenseRecon_with_comm(self):
img, mps, ksp = self.shepp_logan_setup()
lamda = 0
comm = sp.Communicator()
ksp = ksp[comm.rank::comm.size]
mps = mps[comm.rank::comm.size]
img_rec = app.SenseRecon(
ksp, mps, lamda, comm=comm, alg_name='ConjugateGradient',
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
img_rec = app.SenseRecon(
ksp, mps, lamda, alg_name='GradientMethod',
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
img_rec = app.SenseRecon(
ksp,
mps,
lamda,
alg_name='PrimalDualHybridGradient',
max_iter=1000,
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
def test_shepp_logan_SenseRecon(self):
img, mps, ksp = self.shepp_logan_setup()
lamda = 0
img_rec = app.SenseRecon(ksp,
mps,
lamda,
alg_name='ConjugateGradient',
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
img_rec = app.SenseRecon(ksp,
mps,
lamda,
alg_name='GradientMethod',
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
img_rec = app.SenseRecon(ksp,
mps,
lamda,
alg_name='PrimalDualHybridGradient',
max_iter=1000,
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-3, rtol=1e-3)
def test_shepp_logan_SenseRecon(self):
img, mps, ksp = self.shepp_logan_setup()
lamda = 0
for solver in ['ConjugateGradient',
'GradientMethod',
'PrimalDualHybridGradient',
'ADMM']:
with self.subTest(solver=solver):
img_rec = app.SenseRecon(
ksp, mps, lamda, solver=solver,
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-2, rtol=1e-2)
def test_shepp_logan_SenseRecon_with_comm(self):
img, mps, ksp = self.shepp_logan_setup()
lamda = 0
comm = sp.Communicator()
ksp = ksp[comm.rank::comm.size]
mps = mps[comm.rank::comm.size]
for solver in ['ConjugateGradient',
'GradientMethod',
'PrimalDualHybridGradient',
'ADMM']:
with self.subTest(solver=solver):
img_rec = app.SenseRecon(
ksp, mps, lamda, comm=comm, solver=solver,
show_pbar=False).run()
npt.assert_allclose(img, img_rec, atol=1e-2, rtol=1e-2)
