Skip to content

Latest commit

 

History

History
32 lines (20 loc) · 932 Bytes

README.md

File metadata and controls

32 lines (20 loc) · 932 Bytes

SAM-LST

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

We will update the code progressively.

overview

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

Acknowledgements

This repo is based on SAMed and TransUnet.

Thanks original authors for their work!