zhangpeihao / SAM-LST

Pytorch implementation of paper Ladder Fine-tuning approach for SAM integrating complementary network.

SAM-LST

Pytorch implementation of paper Ladder Fine-tuning approach for SAM integrating complementary network.

We will update the code progressively.

Data Preparation

We borrow the data process from SAMed.

Training & Testing

Training

python train.py --root_path <Your folder> --output <Your output path> --warmup --AdamW 

Testing

python test.py --is_savenii --output_dir <Your output directory> --my_ckpt <ckpt path>

Pretrained Models

• Pretrained Models are available at https://drive.google.com/file/d/1sjZhgp02j4Pp_Uvk5Bhg50bELafWu-sR/view?usp=drive_link.

Acknowledgements

This repo is based on SAMed and TransUnet.

Thanks original authors for their work!