FOMO-MRI/AMAES_resenc_b

ResEnc UNet pretrained on FOMO300K using AMAES

This repository contains the checkpoint of the model from the FOMO300K paper.

• Network: Resenc Unet B with 102M parameters (model definition)
• Pretext Task: AMAES.
• Dataset: FOMO300K.
• Code: Asparagus.

Finetuning the model for Segmentation tasks

To finetune the model with the configs used in the paper, simply use the following command

asp_finetune_seg \
    --config-name projects/datapaper/finetune/resenc_[dataset]_fewshot.yaml \
    checkpoint_path=path/to/checkpoint.ckpt

where [dataset] is one of {cerebrum, wmh, isles, atlas, sbm}.

Installing Asparagus

For instructions on how to install asparagus, preprocess your data, and make your own configs, please refer to the Asparagus Readme.