Fixed L1WaveletRecon L1Reg shape.

mikgroup · Aug 22, 2018 · 9c04744 · 9c04744
1 parent fdbfe23
commit 9c04744
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sigpy/mri/app.py b/sigpy/mri/app.py
@@ -146,7 +146,7 @@ def __init__(self, ksp, mps, lamda,
         self.img = sp.util.zeros(mps.shape[1:], dtype=ksp.dtype, device=device)
 
         W = sp.linop.Wavelet(A.ishape, wave_name=wave_name)
-        proxg = sp.prox.UnitaryTransform(sp.prox.L1Reg(A.ishape, lamda), W)
+        proxg = sp.prox.UnitaryTransform(sp.prox.L1Reg(W.oshape, lamda), W)
 
         def g(input):
             device = sp.util.get_device(input)
@@ -195,7 +195,7 @@ def __init__(
         A = linop.Sense(mps, coord=coord)
         self.img = sp.util.zeros(mps.shape[1:], dtype=ksp.dtype, device=device)
         W = sp.linop.Wavelet(A.ishape, wave_name=wave_name)
-        proxg = sp.prox.UnitaryTransform(sp.prox.L1Reg(A.ishape, 1), W)
+        proxg = sp.prox.UnitaryTransform(sp.prox.L1Reg(W.oshape, 1), W)
 
         super().__init__(A, ksp, self.img, proxg, eps, weights=weights, **kwargs)