Mercurial Hosting > traffic-intelligence
diff trafficintelligence/moving.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 | 69b531c7a061 |
children | 5654c9173548 |
line wrap: on
line diff
--- a/trafficintelligence/moving.py Mon Jul 03 10:33:20 2023 -0400 +++ b/trafficintelligence/moving.py Wed Jul 12 12:12:37 2023 -0400 @@ -1376,8 +1376,8 @@ 'truck', 'automated'] -coco2UserTypes = {0: 2, 1: 4, 2: 1, 5: 5, 7: 6} -cocoUserTypeNames = {0: 'person', +coco2Types = {0: 2, 1: 4, 2: 1, 5: 5, 7: 6} +cocoTypeNames = {0: 'person', 1: 'bicycle', 2: 'car', 3: 'motorcycle', @@ -2073,7 +2073,7 @@ self.classifyUserTypeHoGSVMAtInstant(images[t], t, homography, width, height, px, py, minNPixels, rescaleSize, orientations, pixelsPerCell, cellsPerBlock) # compute P(Speed|Class) if speedProbabilities is None or self.aggregatedSpeed < minSpeedEquiprobable: # equiprobable information from speed - userTypeProbabilities = {userType2Num['car']: 1., userType2Num['pedestrian']: 1., userType2Num['bicycle']: 1.} + userTypeProbabilities = {userType2Num['car']: 1., userType2Num['pedestrian']: 1., userType2Num['cyclist']: 1.} else: userTypeProbabilities = {userType2Num[userTypename]: speedProbabilities[userTypename](self.aggregatedSpeed) for userTypename in speedProbabilities} # compute P(Class|Appearance)