MIR.models.Selfsupervised_Learning_Heads

Self-supervised learning heads for registration models.

Classes

SSLHead1Lvl(*args, **kwargs)	Self-supervised learning head with one level.
SSLHeadNLvl(*args, **kwargs)	Self-supervised learning head with multiple levels.
SSL_blocks(*args, **kwargs)	Decoder block for SSL heads.
flow_blocks(*args, **kwargs)	Predict mean (and optionally log-variance) flow fields.