HIT LAB

🧠 DeepIGeoS

Implementation the DeepIGeoS Paper

:page_facing_up: DeepIGeoS: A Deep Interactive Geodesic Framework for Medical Image Segmentation (2018)

additional page : Notion page (in Korean :kr:)

🔍 Prerequisites

Please check environments and requirements before you start. If required, we recommend you to either upgrade versions or install them for smooth running.

☺︎ Environments

Ubuntu 16.04Python 3.7.11

☺︎ Requirements

dotmap
GeodisTK
opencv-python
tensorboard
torch
torchio
torchvision
tqdm
PyQt5

🎞 Datasets

Download the BraTS 2021 dataset form web BraTS 2021 using load_datasets.sh.

$ bash load_datasets.sh

💻 Train

☺︎ P-Net

$ python train_pnet.py -c configs/config_pnet.json

☺︎ R-Net

$ python train_rnet.py -c configs/config_rnet.json

☺︎ Tensorboard