reghbali / kspace-segmentation

Attention-based networks for brain segmentation in k-space

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

🌌 kseg

K-space segmentation models

PyTorch TorchIO Lightning AI Ray Tune

Installation

Clone the repo

git clone git@github.com:rauschecker-sugrue-labs/kspace-segmentation.git
cd kspace-segmentation

Then run (we recommend using a python virtual environment)

pip install --upgrade pip
pip install --upgrade setuptools
pip install -e .

Training

Run

kseg train [model] [data]

Create documentation

To generate the documentation, make sure sphinx is installed and run

cd docs
make html

If you added a new module to the kseg project, you have to update the .rst files in the docs/source/ directory. You can use

cd kspace-segmentation
sphinx-apidoc -F -e -o docs/source kseg

About

Attention-based networks for brain segmentation in k-space


Languages

Language:Python 88.7%Language:Jupyter Notebook 10.5%Language:Shell 0.8%