MIR.models.atlas_builder

Classes

MeanStream(*args, **kwargs)

Mean stream for the Deformable Template Network :param cap: Cap for the mean :param in_shape: Input shape

TemplateCreation(*args, **kwargs)

Deformable Template Network :param reg_model: Registration model :param img_size: Image size :param mean_cap: Mean cap for the MeanStream :param use_sitreg: Use SITReg-style mapping outputs when True; VFA, TransMorph, VoxelMorph-style otherwise :param mode: SpatialTransformer interpolation mode (SITReg path)