SAM-for-BTCV

SAM for BTCV dataset

Setup

./install.sh; requirements in env.yml

Run

cfg.py is all you need. For validation: validation.py and for train: train.py

validation可使用(注意use_box等参数需根据所用模型修改) python validation.py -net_scale='l' -gpu_device=0 -net_ckpt='checkpint路径 -net='sam_pretrain' -use_box=True -vis_image=True

参考checkpoint可见于https://huggingface.co/Zhancun/SAM-for-BTCV,有验证需求可以提出issue,会更新checkpoint