-
Notifications
You must be signed in to change notification settings - Fork 42
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Implement recon models #123
Conversation
Implement varnet
This is addressing #116. Can you look into pylint? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some initial comments.
assert_complex(coil_data, complex_last=True) | ||
assert_complex(sensitivity_map, complex_last=True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this assert_complex
still needed? Can't our internal representation completely switch to complex tensors, and use .is_complex()
or so (that can be done in the assert_complex
of course)?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This will require to change the whole software & models so let's keep this as a future change
# 2D data is batched and contains keys: | ||
# "filename_slice", "slice_no" | ||
# "sampling_mask" of shape: (batch, 1, height, width, 1) | ||
# "sensitivity_map" of shape: (batch, coil, height, width, complex=2) | ||
# "target" of shape: (batch, height, width) | ||
# "masked_kspace" of shape: (batch, coil, height, width, complex=2) | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It feels we should write this down somewhere. Maybe in the documentation?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Documentation of engine?
Rebase from main
…t into implement-recon-models
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. We need some small changes for v0.3, but good for now.
Reconstruction Models added:
Additional models added: