line_actor = fvtk.line(streamlines, colors,
opacity=opacity)
fvtk.add(ren, line_actor)
if r is None:
fvtk.show(ren)
else:
return ren
if __name__ == '__main__':
from load_data import get_train_gt_fibers
streamlines, radii = get_train_gt_fibers()
show_gt_streamlines(streamlines, radii, 'orient', None)
from load_data import get_train_rois
rois, affine = get_train_rois()
streamlines = [s + np.array([24.5, 24.5, 24.5]) for s in streamlines]
from conn_mat import connectivity_matrix
mat, srois, ratio = connectivity_matrix(streamlines, rois)
np.save('train_connmat.npy', mat)
