metadata
license: cc-by-nc-sa-4.0
datasets:
- FOMO-MRI/FOMO300K
pipeline_tag: image-segmentation
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.