shentong-hbu/TriNerSeg

Segmentation of Trigeminal Nerve and Cerebrovasculature in MR-Angiography Images

TriNerSeg

Segmentation of Trigeminal Nerve and Cerebrovasculature in MR-Angiography Images

• The model is defined in model/coarse2fine.py
• train3dv2.py to train network
• predict3d.py to predict segmentations

Requirements

• TorchIO
• SimpleITK
• PyTorch >= 1.5.0