Neighborhood-Attention-Transformer
Neighborhood-Attention-Transformer copied to clipboard
For 3D segmentation
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.
Closing this due to inactivity. We recently integrated our new Fused Neighborhood Attention kernel in NATTEN, which supports 1D, 2D and 3D.