2D reconstructor using CS method and ifft #10
Conversation
There was a problem hiding this comment.
Looks promising :)
Could you cleanup the debugging comments please ?
Also, I think some (light) refactoring could be done to reduce the number of if data_loader.slice_2d call.
A potentially more durable solution would to create a "Slice2D dataloader" that adapt the problem from a N_frames x (N_slices x ...) to (N_slices x N_frammes) x ... And that has the proper shape as well. This would likely fit nicely with what Albane is doing :)
src/snake/mrd_utils/loader.py
Outdated
| """Load slice MRD files k-space frames iteratively.""" | ||
|
|
There was a problem hiding this comment.
Add a bit more documentation about this, mention that it augments a base data_loader for instance, and what it modifies compares to it.
src/snake/mrd_utils/loader.py
Outdated
| traj = traj.reshape(n_acq_per_frame, -1, 3) | ||
| data = data.reshape(self.frame.n_coils, n_acq_per_frame, -1) | ||
|
|
||
| return traj[idx%self.frame.n_shots,:,:2], data[:,idx%self.frame.n_shots,:] |
There was a problem hiding this comment.
| return traj[idx%self.frame.n_shots,:,:2], data[:,idx%self.frame.n_shots,:] | |
| shot_loc = idx%self.frame.n_shots | |
| return traj[shot_loc,:,:2], data[:,shot_loc,:] |
| if data_loader.n_coils > 1: | ||
| smaps = data_loader.get_smaps().squeeze() | ||
| else: | ||
| smaps = None |
| if isinstance(method, str): | ||
| method = get_density(method) | ||
| if not callable(method): | ||
| raise ValueError(f"Unknown density method: {method}") | ||
|
|
There was a problem hiding this comment.
no need for the error raising here, MRI-NUFFT will do it on its own.
| if smaps is not None: | ||
| nufft_operator.smaps = smaps[...,i % data_loader.frame.n_shots] | ||
|
|
alineyyy
force-pushed
the
2d-reconstructors
branch
from
06a0716 to
f6198ca
Compare
2 participants
This makes it possible to reconstruct 2D slices independently using CS method and ifft in
pysap.pywhendata_loader.slice_2d = True.