import nibabel as ni
import osmosis.model.analysis as oza
import osmosis.model.sparse_deconvolution as ssd
import osmosis.model.dti as dti
import osmosis.viz.maya as viz
import osmosis.wm_mask as wm_mask
import os
import osmosis as oz
import osmosis.io as oio
oio.data_path = os.path.join(oz.__path__[0], 'data')
subject = 'SUB2'
data_1k_1, data_1k_2 = oio.get_dwi_data(1000, subject)
seg_path = '%s/%s/%s_t1_class.nii.gz'%(oio.data_path, subject, subject)
dwi = dti.TensorModel(*data_1k_1, params_file='temp')
out_path = '%s/%s/%s_wm_mask.nii.gz'%(oio.data_path, subject, subject)
Loading from file: /home/arokem/usr/lib/python2.7/site-packages/osmosis/data/SUB2/SUB2_b1000_1.bvals Loading from file: /home/arokem/usr/lib/python2.7/site-packages/osmosis/data/SUB2/SUB2_b1000_1.bvecs
wm_mask.make_wm_mask(seg_path, dwi, out_path)
Loading from file: /home/arokem/usr/lib/python2.7/site-packages/osmosis/data/SUB2/SUB2_b1000_1.nii.gz Fitting TensorModel params using dipy
vol_anat = oio.get_t1(subject, resample=ni.load(oio.data_path + '/%s/%s_wm_mask.nii.gz'%(subject, subject)))
vol_mask = ni.load(out_path).get_data()
vol_mask[vol_mask<1] = np.nan
%gui wx
viz.plot_cut_planes(vol_anat,
overlay=vol_mask,
overlay_cmap="RdYlGn",
slice_coronal=40,
slice_saggital=15,
slice_axial=30,
view_azim=-40,
view_elev=60)
Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x96d6710> Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x96e1a10> Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x9b3bd10> Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x9b45dd0> Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x9b5c350> Cannot set extents for <mayavi.modules.image_plane_widget.ImagePlaneWidget object at 0x9b67650>
<mayavi.core.scene.Scene at 0x96cddd0>