Spine Segmentation

Description

Spine segmentation using SwinUNETR on MRI images.

# clone project
git clone https://github.com/tiendzung/spine-segmentation
cd spine-segmentation

Train model with default configuration

# train on CPU
python src/train.py trainer=cpu

# train on GPU
python src/train.py trainer=gpu

Train model with chosen experiment configuration from configs/experiment/

python src/train.py experiment=experiment_name.yaml

You can override any parameter from command line like this

python src/train.py trainer.max_epochs=20 data.batch_size=64

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
.github		.github
configs		configs
notebooks		notebooks
outputs		outputs
scripts		scripts
src		src
tests		tests
visualize		visualize
.DS_Store		.DS_Store
.dockerignore		.dockerignore
.env.example		.env.example
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
.project-root		.project-root
Dockerfile		Dockerfile
Makefile		Makefile
README.md		README.md
config.sh		config.sh
container.sh		container.sh
docker.sh		docker.sh
environment.yaml		environment.yaml
image-1.png		image-1.png
image.png		image.png
pyproject.toml		pyproject.toml
requirements.txt		requirements.txt
setup.py		setup.py