Mercurial Hosting > traffic-intelligence
comparison python/cvutils.py @ 236:eb4525853030
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()) |