MIR.models.MultiMorph

MultiMorph model Abulnaga, S. M., Hoopes, A., Dey, N., Hoffmann, M., Fischl, B., Guttag, J., & Dalca, A. (2025). MultiMorph: On-demand Atlas Construction. In Proceedings of the Computer Vision and Pattern Recognition Conference (pp. 30906-30917).

Code retrieved from: https://github.com/mabulnaga/multimorph

Modified and tested by: Junyu Chen jchen245@jhmi.edu Johns Hopkins University

Functions

make_epoch_batches(n_samples, min_bs, max_bs)

Create randomized batch index lists for one epoch.

Classes

ComposeCentrality(*args, **kwargs)
DeformationFieldComposer(*args, **kwargs)
FastMeanConv3d(*args, **kwargs)	Perform a group mean convolution (see UniverSeg paper).
FastMeanConv3dUp(*args, **kwargs)	Perform a group mean convolution for the upsampling UNet.
GroupConv2d(*args, **kwargs)	Perform a group convolution without communication.
GroupConv3d(*args, **kwargs)	Perform a group convolution without communication.
GroupNet(*args, **kwargs)
GroupNet3D(*args, **kwargs)
Lambda(*args, **kwargs)	Wrap a callable inside an nn.Module.
ListBatchSampler(*args, **kwargs)	Yield precomputed lists of indices as batches.
MaxPool2d(*args, **kwargs)
MaxPool3d(*args, **kwargs)
MeanConv2d(*args, **kwargs)	Perform a group mean convolution (see UniverSeg paper).
MeanConv3d(*args, **kwargs)	Perform a group mean convolution (see UniverSeg paper).
ResizeTransform(*args, **kwargs)	Resize and rescale a vector field transform.
SimpleUNet(*args, **kwargs)
SubtractMean(*args, **kwargs)	Subtract mean along a specified dimension.
UpsamplingBilinear2d(*args, **kwargs)
UpsamplingTrilinear3d(*args, **kwargs)
VecInt(*args, **kwargs)	Integrate a vector field via scaling and squaring.
VecIntGroup(*args, **kwargs)	Vector integration with group dimension
Warp2d(*args, **kwargs)
Warp3d(*args, **kwargs)