Skip to content
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

standalone ICON and GradICON losses for registration network training #80

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

BailiangJ
Copy link

Hi @HastingsGreer ,

Thanks a lot for sharing the code.

I have created the standalone ICON and GradICON losses with Pytorch, which can be directly used in general registration network training. Modification would be much simpler to plug in and play with them.

Relevant to issue #79

Double check is needed to see if there is any error in the implementation. :)

Cheers,
Bailiang

@BailiangJ
Copy link
Author

Since the flow field in the official code is normalized to [-1,1] while the implementation here is the displacement in the original resolution, epsilon = torch.randn_like(forward_flow) * (1.0 / self.image_size[-1]) in line 313,382 and delta in GradICONLoss might need to be changed accordingly.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant