Mercurial > hg > nsaunier > traffic-intelligence

comparison python/cvutils.py @ 236:eb4525853030

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added script to display trajectories
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 06 Jul 2012 01:03:52 -0400
parents 584613399513
children e0988a8ace0c
comparison
equal deleted inserted replaced
235:584613399513 236:eb4525853030
141 ret, img = capture.read() 141 ret, img = capture.read()
142 if ret: 142 if ret:
143 print('frame {0}'.format(frameNum)) 143 print('frame {0}'.format(frameNum))
144 for obj in objects: 144 for obj in objects:
145 if obj.existsAtInstant(frameNum): 145 if obj.existsAtInstant(frameNum):
146 if obj.getFirstInstant() == frameNum: 146 if not hasattr(obj, 'projectedPositions'):
147 if homography != None: 147 if homography != None:
148 obj.projectedPositions = obj.positions.project(homography) 148 obj.projectedPositions = obj.positions.project(homography)
149 else: 149 else:
150 obj.projectedPositions = obj.positions 150 obj.projectedPositions = obj.positions
151 draw(img, obj.projectedPositions, cvRed, frameNum-obj.getFirstInstant()) 151 draw(img, obj.projectedPositions, cvRed, frameNum-obj.getFirstInstant())