Skip to content

Extending MaxViT (Multi-Axis Vision Transformer) to 3D Space

License

Notifications You must be signed in to change notification settings

nengwp/3D-MaxViT-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

3D-MaxViT-pytorch

Extending MaxViT (Multi-Axis Vision Transformer) to 3D Space

MaxViT

MaxViT:Multi-Axis Vision Transformer. In this paper author introduce an efficient and scalable attention model , call multi-axis attention, which consists of two aspects: blocked local and dilated global attention.


max-sa

3D MaxViT

We extend MaxViT to 3D space to accommodate medical images such as CT and MRI. We implemented 3D Max-SA(3D Multi-axis self-attention) as shown in the figure to facilitate insertion into existing medical networks such as 3DUnet.


3d-max-sa

Acknowledgement

We refer to part of the code of vit-pytorch/maxvit

About

Extending MaxViT (Multi-Axis Vision Transformer) to 3D Space

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages