diff scripts/classify-objects.py @ 682:fbe29be25501 dev

corrected bug
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 05 Jun 2015 02:35:29 +0200
parents da1352b89d02
children f3a0b652b590
line wrap: on
line diff
--- a/scripts/classify-objects.py	Fri Jun 05 02:25:50 2015 +0200
+++ b/scripts/classify-objects.py	Fri Jun 05 02:35:29 2015 +0200
@@ -46,6 +46,10 @@
                       'pedestrian': lambda s: norm(params.meanPedestrianSpeed, params.stdPedestrianSpeed).pdf(s), 
                       'bicycle': lambda s: lognorm(params.scaleCyclistSpeed, loc = 0., scale = np.exp(params.locationCyclistSpeed)).pdf(s)} # lognorm shape, loc, scale
 
+def plotSpeedProbabilities():
+    for k in speedProbabilities:
+        plot(arange(0.1, 50, 0.1), [speedProbabilities[k](s/3.6/25) for s in arange(0.1, 50, 0.1)])
+
 objects = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'object')
 features = storage.loadTrajectoriesFromSqlite(params.databaseFilename, 'feature')
 intervals = []