Name		Name	Last commit message	Last commit date
parent directory ..
examples		examples
hooks		hooks
images		images
model		model
utils		utils
.gitignore		.gitignore
Dockerfile		Dockerfile
LICENSE		LICENSE
README.md		README.md
download_dataset.py		download_dataset.py
download_script.sh		download_script.sh
export.py		export.py
main.py		main.py
readme_origin.md		readme_origin.md
requirements.txt		requirements.txt
run_get_hippocampus_data.sh		run_get_hippocampus_data.sh
run_training.sh		run_training.sh

README.md

VNet

Model description

V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation Fausto Milletari, Nassir Navab, Seyed-Ahmad Ahmadi

Prepare

pip3 install -r requirements.txt

Download dataset

python3  download_dataset.py --data_dir ./data

Run training

single card

python3 examples/vnet_train_and_evaluate.py --gpus 1 --batch_size 8 --base_lr 0.0001 --data_dir ./data/Task04_Hippocampus/ --model_dir ./model_train/

8 cards

python3 examples/vnet_train_and_evaluate.py --gpus 8 --batch_size 8 --base_lr 0.0001 --data_dir ./data/Task04_Hippocampus/ --model_dir ./model_train/

Result

	background_dice	anterior_dice	posterior_dice
multi_card	0.9912699	0.83743376	0.81537557