SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

For 3D segmentation

LambdaLi opened this issue · comments

Your work is breathtaking, I asked a naive question, can it be used in 3D image segmentation now? I only know 1D and 2D in your natten, are you still developing?

Thank you for your interest. There is a basic 3D NA in NATTEN right now, you'd just need to build it from source. However, it's a very limited implementation, so I wouldn't expect much from it performance wise.

And yes, NATTEN development is still ongoing, and we're going to have more releases coming up soon.