MIR.models.Swin_Transformer

Swin Transformer backbone for 3D medical image registration.

Functions

window_partition(x, window_size)
window_reverse(windows, window_size, H, W, L)

Classes

BasicLayer(*args, **kwargs)	A basic Swin Transformer layer for one stage.
Mlp(*args, **kwargs)	Feed-forward MLP block used inside transformer blocks.
PatchEmbed(*args, **kwargs)	Image to Patch Embedding :param patch_size: Patch token size.
PatchMerging(*args, **kwargs)	Patch Merging Layer.
SinPositionalEncoding3D(*args, **kwargs)
SinusoidalPositionEmbedding(*args, **kwargs)	Rotary Position Embedding
SwinTransformer(*args, **kwargs)	Swin Transformer
SwinTransformerBlock(*args, **kwargs)	Swin Transformer Block.
WindowAttention(*args, **kwargs)	Window based multi-head self attention (W-MSA) module with relative position bias.