18 Star 82 Fork 49

DeepSpark/DeepSparkHub

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
README.md 978 Bytes
一键复制 编辑 原始数据 按行查看 历史
majorli6 提交于 2023-03-21 15:35 . make model name in README.md the same.

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
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/deep-spark/deepsparkhub.git
git@gitee.com:deep-spark/deepsparkhub.git
deep-spark
deepsparkhub
DeepSparkHub
master

搜索帮助

A270a887 8829481 3d7a4017 8829481