Skip to content

Commit

Permalink
Update vimunet.md (#246)
Browse files Browse the repository at this point in the history
  • Loading branch information
anwai98 authored May 5, 2024
1 parent 112960c commit d09d045
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vimunet.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ This is the documentation for the installation instructions, known issues and li
## TLDR
1. Please install [`torch-em`](https://github.com/constantinpape/torch-em) and `ViM` (based on our fork: https://github.com/anwai98/Vim)
2. Supports `ViM Tiny` and `ViM Small` for 2d segmentation using ViM-UNet.
3. Check out [our preprint](https://arxiv.org/abs/2404.07705) for more details.
3. Check out [our preprint](https://arxiv.org/abs/2404.07705) (accepted at [MIDL 2024 - Short Paper](https://openreview.net/forum?id=PYNwysgFeP)) for more details.
- The key observation: "ViM-UNet performs similarly or better that UNet (depending on the task), and outperforms UNETR while being more efficient." Its main advantage is for segmentation problems that rely on larger context.

## Benchmarking Methods
Expand Down

0 comments on commit d09d045

Please sign in to comment.