Mercurial Hosting > traffic-intelligence
diff scripts/classify-objects.py @ 1228:5654c9173548
merged (bicycle)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 12 Jul 2023 13:21:08 -0400 |
parents | d478d3122804 |
children | bb14f919d1cb |
line wrap: on
line diff
--- a/scripts/classify-objects.py Wed Jul 12 13:14:09 2023 -0400 +++ b/scripts/classify-objects.py Wed Jul 12 13:21:08 2023 -0400 @@ -37,9 +37,9 @@ pedNorm = norm(classifierParams.meanPedestrianSpeed, classifierParams.stdPedestrianSpeed) # numpy lognorm shape, loc, scale: shape for numpy is scale (std of the normal) and scale for numpy is exp(location) (loc=mean of the normal) bicLogNorm = lognorm(classifierParams.scaleCyclistSpeed, loc = 0., scale = np.exp(classifierParams.locationCyclistSpeed)) -speedProbabilities = {'car': lambda s: carNorm.pdf(s), - 'pedestrian': lambda s: pedNorm.pdf(s), - 'bicycle': lambda s: bicLogNorm.pdf(s)} +speedProbabilities = {moving.userTypeNames[1]: lambda s: carNorm.pdf(s), + moving.userTypeNames[2]: lambda s: pedNorm.pdf(s), + moving.userTypeNames[4]: lambda s: bicLogNorm.pdf(s)} if args.plotSpeedDistribution: import matplotlib.pyplot as plt