forked from juhuntenburg/myelinconnect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
subcortex_masks.py
33 lines (19 loc) · 863 Bytes
/
subcortex_masks.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import pandas as pd
from nipype.interfaces.freesurfer import Binarize
from nipype.interfaces.io import SelectFiles
subjects = pd.read_csv('/scr/ilz3/myelinconnect/subjects.csv')
subjects=list(subjects['DB'])
subjects.remove('KSMT')
labels= [11, 12, 13, 16, 18] + range(30,42)
templates={'seg': '/scr/ilz3/myelinconnect/struct/seg/{subject}*seg_merged.nii.gz'}
mask_file = '/scr/ilz3/myelinconnect/struct/myelinated_thickness/subcortex_mask/%s_subcortical_mask.nii.gz'
for subject in subjects:
select = SelectFiles(templates)
select.inputs.subject = subject
select.run()
seg_file = select.aggregate_outputs().seg
binarize = Binarize(match = labels,
out_type = 'nii.gz')
binarize.inputs.binary_file = mask_file%subject
binarize.inputs.in_file=seg_file
binarize.run()