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

Update UNETR to enable resize to longest side #192

Merged
merged 4 commits into from
Jan 1, 2024
Merged

Conversation

constantinpape
Copy link
Owner

to enable consistency with SAM and use full encoder input shape rather than naively padding.

This is still a draft to discuss what the best default is and how we can best integrate this with previous code.

torch_em/model/unetr.py Outdated Show resolved Hide resolved
@constantinpape constantinpape marked this pull request as ready for review January 1, 2024 18:30
@constantinpape
Copy link
Owner Author

I went ahead and set resize as the default and also updated the functionality so that we don't need segment-anything for it. I will merge once the tests pass.

@anwai98
Copy link
Contributor

anwai98 commented Jan 1, 2024

Awesome. Thanks Constantin!

@constantinpape constantinpape merged commit fad7fb6 into main Jan 1, 2024
4 checks passed
@constantinpape constantinpape deleted the unetr-updates branch January 1, 2024 18:38
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.

None yet

2 participants