Hippocampal segmentation using the UNet network for each axis, inspired by https://github.com/MICLab-Unicamp/e2dhipseg
- Red: False Positive
- Green: True Positive
- Blue: False Negative
git clone https://github.com/aguirrejuan/hipposeg.git
pip install -r requirements.txt
Download the Weigths trained with ADNI HarP Dataset and put in the model directory.
python inference.py --path_vol "example.nii"
The result will be in the current directory
Download the VGG11 model and put in the model directory.
python train.py --train_path "directory_mris/" \
--train_paht_label "directory_mris_labels/" \
--epochs number_of_epochs
python evaluate.py --path_data "directory_mris/" \
--path_data_label "directory_mris_labels/"