Slide-SAM/readme.md
2023-12-12 13:43:26 +08:00

358 B

Slide-SAM: Medical SAM meets sliding window

Before Training

install tutils

pip install trans-utils

prepare datasets

python -m  datasets.generate_txt

cache 3d data into slices

python -m datasets.cache_datasets3d

Training

run training

CUDA_VISIBLE_DEVICES=0,1,2,3 python -m core.ddp --tag debug