Mercurial > hg > nsaunier > traffic-intelligence

comparison python/prediction.py @ 948:584b9405e494

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added safety analysis parameters for motion patterns
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 21 Jul 2017 12:11:55 -0400
parents e5970606066f
children d6c1c05d11f5
comparison
equal deleted inserted replaced
947:053484e08947 948:584b9405e494
581 obj.currentPredictionFeatures[:] = [f for f in obj.currentPredictionFeatures if f.existsAtInstant(instant)] 581 obj.currentPredictionFeatures[:] = [f for f in obj.currentPredictionFeatures if f.existsAtInstant(instant)]
582 firstInstants = [(f,f.getFirstInstant()) for f in obj.getFeatures() if f.existsAtInstant(instant) and f not in obj.currentPredictionFeatures] 582 firstInstants = [(f,f.getFirstInstant()) for f in obj.getFeatures() if f.existsAtInstant(instant) and f not in obj.currentPredictionFeatures]
583 firstInstants.sort(key = lambda t: t[1]) 583 firstInstants.sort(key = lambda t: t[1])
584 for f,t1 in firstInstants[:min(self.nPredictedTrajectories, len(firstInstants), self.nPredictedTrajectories-len(obj.currentPredictionFeatures))]: 584 for f,t1 in firstInstants[:min(self.nPredictedTrajectories, len(firstInstants), self.nPredictedTrajectories-len(obj.currentPredictionFeatures))]:
585 obj.currentPredictionFeatures.append(f) 585 obj.currentPredictionFeatures.append(f)
586 print len(obj.currentPredictionFeatures), self.nPredictedTrajectories
587 for f in obj.currentPredictionFeatures: 586 for f in obj.currentPredictionFeatures:
588 self.addPredictedTrajectories(predictedTrajectories, f, instant) 587 self.addPredictedTrajectories(predictedTrajectories, f, instant)
589 else: 588 else:
590 self.addPredictedTrajectories(predictedTrajectories, obj, instant) 589 self.addPredictedTrajectories(predictedTrajectories, obj, instant)
591 return predictedTrajectories 590 return predictedTrajectories