search

The-AI-Summer / self-attention-cv

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.
https://theaisummer.com/
MIT License
1.18k stars 154 forks source link

Request for Including UNETR #6

Closed Siyuan89 closed 3 years ago

Siyuan89 commented 3 years ago

Thanks for great work ! I noticed nice implementation of this paper (https://arxiv.org/abs/2103.10504) here:

https://github.com/tamasino52/UNETR/blob/main/unetr.py

It would be great if this can also be included in your repo, since it comes with lots of other great features. So we can explore more.

Thanks ~

black0017 commented 3 years ago

Interesting work! I am interested in reimplementing/adjusting the official repo in this one.

From a first pass I noticed that they do not used any pretraining, that sounds very interesting!

Thanks for letting me know about this work! @Siyuan89

black0017 commented 3 years ago

added on v1.2 @Siyuan89

Siyuan89 commented 3 years ago

Thank you @black0017 for your work. Also any working example of how to use it ?