def link_ridges(vec, search_range, memory=0, **kwargs):
# generate point levels from the previous steps
levels = [[Point1D_circ(q, phi, v) for phi, v in pks] for q, pks in vec]
trks = pt.link_full(levels,
2 * np.pi,
search_range,
hash_cls=hash_line_angular,
memory=memory,
track_cls=lf_Track)
for t in trks:
t.classify2(**kwargs)
trks.sort(key=lambda x: x.phi)
return trks
def link_ridges(vec, search_range, memory=0, **kwargs):
# generate point levels from the previous steps
levels = [[Point1D_circ(q, phi, v) for phi, v in pks] for q, pks in vec]
trks = pt.link_full(levels,
2 * np.pi,
search_range,
hash_cls=hash_line_angular,
memory=memory,
track_cls=lf_Track)
for t in trks:
t.classify2(**kwargs)
trks.sort(key=lambda x: x.phi)
return trks
def _track(self):
self._links = tracking.link_full(self._levels,
self._frameDimensions,
self._searchRadius,
tracking.Hash_table,
self._memory)
intlinex=[float(i) for i in templinex]
intliney=[float(i) for i in templiney]
for index in range(len(intlinex)):
if intlinex[index]==0.00 and intliney[index]==0.00:
continue
else:
data[index].append(pt.PointND(index+1,np.asarray((intlinex[index],intliney[index]))))
print "Line "+str(linenumber)+" is done!"
linenumber+=1
#print [intlinex[0],intliney[0]]
#print data[0]
print "Tracking started!"
#print levels
# do the tracking
t = pt.link_full(data[:],(1930,1930),3,pt.Hash_table)
print "Linking is done!"
# plot tracks
fig = plt.figure()
ax = fig.gca()
out=open('trackeddata.txt','w')
out=open('trackeddata.txt','a')
for track in t:
pos=track.points
out.write(str(pos[0].t)+' ')
for p in pos:
out.write(str(p.pos[1])+' '+str(p.pos[0])+' '+str(1.00)+' ')
out.write('\n')
out.close()
