用于3D图像到图像翻译的生成对抗网络
该网络的目的是训练从T1w图像创建T2w图像的模型,反之亦然。 它以完整的3D NIfTI作为输入,并能够直接生成相应模态的3D体积。
该存储库基于Per Welander和Anders Eklund在
训练此网络的最大挑战之一是将其全部存储在内存中。 必须有GPU才能在合理的时间内(不到一周)训练模型,但是这种加速是以RAM为代价的。 为了训练此3D GAN,我们使用了具有32GB RAM的单个NVIDIA Tesla V100。 少了可能会导致内存不足错误。
系统设置
培训是在带有CUDA版本10.2.89的NVIDIA Tesla V100上完成的
所需的python软件包版本保存在requirements.txt中,应安装在虚拟环境中
python3 -m venv env
source env/bin/activate
pip insta
2021-04-22 20:25:05
22.82MB
Python
1