spyrit.core.recon.LearnedPGD.reconstruct
- LearnedPGD.reconstruct(x)[source]
Reconstruction step of a reconstruction network
- Args:
x: raw measurement vectors- Shape:
x: \((BC,2M)\)output: \((BC,1,H,W)\)- Example:
>>> from spyrit.core.meas import HadamSplit2d >>> from spyrit.core.prep import UnsplitRescale >>> from spyrit.core.recon import LearnedPGD
>>> import torch >>> acqu = HadamSplit2d(32, M=400) >>> prep = UnsplitRescale() >>> recnet = LearnedPGD(acqu, prep) >>> y = torch.randn(10, 1, 800) >>> z = recnet.reconstruct(y) >>> print(z.shape) torch.Size([10, 1, 32, 32])