This repo is containing all the code related to the spine foundation model.
To be able to train the model you need to:
- Create a venv using the following command:
python -m venv .env_SpineFoundation - Activate your venv by running:
source .env_SpineFoundation/bin/activate - Install requirements:
pip install requirements.txt - Have a data folder containing datasets following the BIDS convention.
Command to run :
python -m mae_training.build \
--config ./mae_training/config.json \
--ddp (optional)