This code is part of the implementation of our TMI paper Objective Detection of Eloquent Axonal Pathways to Minimize Postoperative Deficits in Pediatric Epilepsy Surgery using Diffusion Tractography and Convolutional Neural Networks. The model for brain fiber classification is included here:
- deep CNN with focal loss + center loss + attention.
-
test: an example .mat file can be found here. The example output folder is provided.
For example,
python /path/to/test_atm.py ./example.mat ./DCNN_CL_ATT.model 65
Remember to change all the paths in the scripts according to your file location. The class indices are shown as below:
