def skeletonize_both():
from dipy.viz import fos
from dipy.core.track_metrics import downsample
from dipy.core.track_performance import local_skeleton_clustering, most_similar_track_mam
froi='/home/eg309/Data/ICBM_Wmpm/ICBM_WMPM.nii'
wI=get_roi(froi,9,0) #4 is genu
fname='/home/eg309/Data/PROC_MR10032/subj_03/101/1312211075232351192010092217244332311282470ep2dadvdiffDSI10125x25x25STs004a001_QA_warp.dpy'
#fname='/home/eg309/Data/PROC_MR10032/subj_03/101/1312211075232351192010092217244332311282470ep2dadvdiffDSI10125x25x25STs004a001_QA_native.dpy'
#fname='/home/eg309/Data/PROC_MR10032/subj_06/101/13122110752323511930000010092916083910900000227ep2dadvdiffDSI10125x25x25STs004a001_QA_native.dpy'
fname2='/home/eg309/Data/PROC_MR10032/subj_06/101/13122110752323511930000010092916083910900000227ep2dadvdiffDSI10125x25x25STs004a001_QA_warp.dpy'
r=fos.ren()
#'''
dpr=Dpy(fname,'r')
T=dpr.read_indexed(range(2*10**4))
dpr.close()
print len(T)
Td=[downsample(t,3) for t in T if length(t)>40]
C=local_skeleton_clustering(Td,d_thr=20.)
for c in C:
#color=np.random.rand(3)
color=fos.red
if C[c]['N']>0:
Ttmp=[]
for i in C[c]['indices']:
Ttmp.append(T[i])
si,s=most_similar_track_mam(Ttmp,'avg')
print si,C[c]['N']
fos.add(r,fos.line(Ttmp[si],color))
dpr=Dpy(fname2,'r')
T=dpr.read_indexed(range(2*10**4))
dpr.close()
print len(T)
Td=[downsample(t,3) for t in T if length(t)>40]
C=local_skeleton_clustering(Td,d_thr=20.)
#r=fos.ren()
for c in C:
#color=np.random.rand(3)
color=fos.yellow
if C[c]['N']>0:
Ttmp=[]
for i in C[c]['indices']:
Ttmp.append(T[i])
si,s=most_similar_track_mam(Ttmp,'avg')
print si,C[c]['N']
fos.add(r,fos.line(Ttmp[si],color))
#'''
fos.add(r,fos.point(wI,fos.green))
fos.show(r)
def skeletonize():
froi='/home/eg309/Data/ICBM_Wmpm/ICBM_WMPM.nii'
wI=get_roi(froi,35,1) #4 is genu
#fname='/home/eg309/Data/PROC_MR10032/subj_03/101/1312211075232351192010092217244332311282470ep2dadvdiffDSI10125x25x25STs004a001_QA_warp.dpy'
#fname='/home/eg309/Data/PROC_MR10032/subj_03/101/1312211075232351192010092217244332311282470ep2dadvdiffDSI10125x25x25STs004a001_QA_native.dpy'
#fname='/home/eg309/Data/PROC_MR10032/subj_06/101/13122110752323511930000010092916083910900000227ep2dadvdiffDSI10125x25x25STs004a001_QA_native.dpy'
fname='/home/eg309/Data/PROC_MR10032/subj_06/101/13122110752323511930000010092916083910900000227ep2dadvdiffDSI10125x25x25STs004a001_QA_warp.dpy'
dpr=Dpy(fname,'r')
T=dpr.read_indexed(range(2*10**4))
dpr.close()
print len(T)
from dipy.core.track_metrics import downsample
from dipy.core.track_performance import local_skeleton_clustering, most_similar_track_mam
Td=[downsample(t,3) for t in T]
C=local_skeleton_clustering(Td,d_thr=20.)
#Tobject=np.array(T,dtype=np.object)
from dipy.viz import fos
r=fos.ren()
#skeleton=[]
for c in C:
color=np.random.rand(3)
if C[c]['N']>0:
Ttmp=[]
for i in C[c]['indices']:
Ttmp.append(T[i])
si,s=most_similar_track_mam(Ttmp,'avg')
print si,C[c]['N']
fos.add(r,fos.line(Ttmp[si],color))
#print len(skeleton)
#fos.add(r,fos.line(skeleton,color))
#fos.add(r,fos.line(T,fos.red))
fos.show(r)
print 'Loading file...'
streams,hdr=tv.read(fname)
print 'Copying tracks...'
T=[i[0] for i in streams]
print 'Representing tracks using only 3 pts...'
tracks=[tm.downsample(t,3) for t in T]
print 'Deleting unnecessary data...'
del streams,hdr
print 'Hidden Structure Clustering...'
now=time.clock()
C=pf.local_skeleton_clustering(tracks,d_thr=20)
print 'Done in', time.clock()-now,'s.'
print 'Reducing the number of points...'
T=[pf.approximate_ei_trajectory(t) for t in T]
print 'Showing initial dataset.'
r=fos.ren()
fos.add(r,fos.line(T,fos.white,opacity=0.1))
fos.show(r)
print 'Showing dataset after clustering.'
fos.clear(r)
colors=np.zeros((len(T),3))
for c in C:
color=np.random.rand(1,3)
print 'Copying tracks...' T = [i[0] for i in streams] T = T[:len(T) / 5] #T=T[:1000] print 'Representing tracks using only 3 pts...' tracks = [tm.downsample(t, 3) for t in T] print 'Deleting unnecessary data...' del streams, hdr print 'Local Skeleton Clustering...' now = time.clock() C = pf.local_skeleton_clustering(tracks, d_thr=20) print 'Done in', time.clock() - now, 's.' print 'Reducing the number of points...' T = [pf.approx_polygon_track(t) for t in T] print 'Showing initial dataset.' #r=fos.ren() #fos.add(r,fos.line(T,fos.white,opacity=0.1)) #fos.show(r) data = T colors = [np.tile(np.array([1, 1, 1, opacity], 'f'), (len(t), 1)) for t in T]
