comparison scripts/classify-objects.py @ 1226:d478d3122804

change of bicycle to cyclist
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 12 Jul 2023 12:12:37 -0400
parents 9f7a4a026dab
children 5654c9173548
comparison
equal deleted inserted replaced
1225:202073959fb4 1226:d478d3122804
37 pedNorm = norm(classifierParams.meanPedestrianSpeed, classifierParams.stdPedestrianSpeed) 37 pedNorm = norm(classifierParams.meanPedestrianSpeed, classifierParams.stdPedestrianSpeed)
38 # 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) 38 # 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)
39 bicLogNorm = lognorm(classifierParams.scaleCyclistSpeed, loc = 0., scale = np.exp(classifierParams.locationCyclistSpeed)) 39 bicLogNorm = lognorm(classifierParams.scaleCyclistSpeed, loc = 0., scale = np.exp(classifierParams.locationCyclistSpeed))
40 speedProbabilities = {'car': lambda s: carNorm.pdf(s), 40 speedProbabilities = {'car': lambda s: carNorm.pdf(s),
41 'pedestrian': lambda s: pedNorm.pdf(s), 41 'pedestrian': lambda s: pedNorm.pdf(s),
42 'bicycle': lambda s: bicLogNorm.pdf(s)} 42 'cyclist': lambda s: bicLogNorm.pdf(s)}
43 43
44 if args.plotSpeedDistribution: 44 if args.plotSpeedDistribution:
45 import matplotlib.pyplot as plt 45 import matplotlib.pyplot as plt
46 plt.figure() 46 plt.figure()
47 speeds = np.arange(0.1, args.maxSpeedDistributionPlot, 0.1) 47 speeds = np.arange(0.1, args.maxSpeedDistributionPlot, 0.1)