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.
FourierNet(*args, **kwargs)	MIR-callable 3-D Fourier-Net registration model.
HyperMorph	HyperMorph: hyperparameter-conditioned VoxelMorph.
HyperTransMorph	Hyper-TransMorph: TransMorph conditioned on a regularization hyperparameter.
HyperVFA(*args, **kwargs)	Hyperparameter-conditioned VFA model.
LessNet(*args, **kwargs)	MIR-callable LessNet registration model.
MultiMorph	MultiMorph group-wise registration model for on-demand atlas construction.
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.
TransMorphLDDMM(*args, **kwargs)	TransMorph with an LDDMM (time-varying velocity field) transformation model.
TransVFA(*args, **kwargs)	TransVFA model for image registration.
VFA(*args, **kwargs)	VFA model for image registration.
VoxelMorph	VoxelMorph: U-Net based deformable registration network.
affine_reg	3D affine registration module using MIR SpatialTransformer.
atlas_builder	Deformable template (atlas) construction: a template-creation wrapper and a running-mean flow stream.
configs_FourierNet	Configuration presets for Fourier-Net models.
configs_LessNet	Configuration presets for LessNet models.
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.
lddmm	Model-agnostic LDDMM (time-varying velocity field) components for MIR.
registration_utils	Model-level registration utilities (transformers and geometry).