Code and trained model for the paper "A machine learning-based method for estimating the number and orientations of major fascicles in diffusion-weighted magnetic resonance imaging".
https://arxiv.org/abs/2006.11117
Files included:
training_proposed.py --> generates simulated training data and trains the proposed method
test_all_methods.py --> tests all methods compared in the paper
dci.py --> includes functions used by the above two scripts
models: include trained MLP weightds for the proposed method.