Update readme.md

This commit is contained in:
Curli-quan 2023-12-12 13:43:26 +08:00 committed by GitHub
parent e04459c6fe
commit 789eb05278
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,8 +1,12 @@
<!-- # Slide-SAM -->
# Slide-SAM: Medical SAM meets sliding window
## Before Training
install tutils
```
pip install trans-utils
```
## Training
prepare datasets
```
python -m datasets.generate_txt
@ -13,6 +17,7 @@ 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