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)