ResEnc UNet pretrained on FOMO300K using AMAES

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

NB: This checkpoint was trained on an earlier version of FOMO300K, which included 11K scans that are no longer part of the dataset.

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}.

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

Downloads last month: -; Downloads are not tracked for this model. How to track