MIR.models

Model zoo and configuration helpers for MIR registration pipelines.

Modules

Deformable_Swin_Transformer	Deformable Swin Transformer backbone for 3D registration.
Deformable_Swin_Transformer_v2	Deformable Swin Transformer v2 backbone for 3D registration.
HyperMorph	HyperMorph
HyperTransMorph	Hyper-TransMorph model Chen, J., Du, Y., He, Y., Segars, W.
HyperVFA(*args, **kwargs)	Hyperparameter-conditioned VFA model.
MultiMorph	MultiMorph model Abulnaga, S.
SITReg
Segmentation_Heads	Segmentation heads and decoder blocks for volumetric models.
Selfsupervised_Learning_Heads	Self-supervised learning heads for registration models.
Swin_Transformer	Swin Transformer backbone for 3D medical image registration.
Synthesis_Heads	Synthesis heads for image reconstruction from features.
TransMorph(*args, **kwargs)	TransMorph model.
TransVFA(*args, **kwargs)	TransVFA model for image registration.
VFA(*args, **kwargs)	VFA model for image registration.
VoxelMorph	VoxelMorph
affine_reg	3D affine registration module using MIR SpatialTransformer.
atlas_builder
configs_TransMorph	Configuration presets for TransMorph models.
configs_VFA	Configuration presets for VFA models.
configs_VoxelMorph	Configuration presets for VoxelMorph models.
convexAdam	ConvexAdam registration wrappers and configuration presets.
registration_utils	Model-level registration utilities (transformers and geometry).