3D网络
用于大脑分割的3D Unet的Chainer实现。 培训配置位于configs / base.yml。 由于GPU内存的限制,我们使用了基于补丁的方法。
要求
SimpleITK
连连看v4
yaml
网络架构
3D Unet架构
用法
训练训练3D unet。
python train.py -h
optional arguments:
-h, --help show this help message and exit
--gpu GPU, -g GPU GPU ID (negative value indicates CPU)
--base BASE, -B BASE base directory path of program files
--config_path CONFIG_PATH
2022-02-24 14:49:22
80.99MB
Python
1