Mercurial Hosting > traffic-intelligence

comparison trafficintelligence/moving.py @ 1226:d478d3122804

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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
comparison
equal deleted inserted replaced
1225:202073959fb4 1226:d478d3122804
1374 'cyclist', 1374 'cyclist',
1375 'bus', 1375 'bus',
1376 'truck', 1376 'truck',
1377 'automated'] 1377 'automated']
1378 1378
1379 coco2UserTypes = {0: 2, 1: 4, 2: 1, 5: 5, 7: 6} 1379 coco2Types = {0: 2, 1: 4, 2: 1, 5: 5, 7: 6}
1380 cocoUserTypeNames = {0: 'person', 1380 cocoTypeNames = {0: 'person',
1381 1: 'bicycle', 1381 1: 'bicycle',
1382 2: 'car', 1382 2: 'car',
1383 3: 'motorcycle', 1383 3: 'motorcycle',
1384 5: 'bus', 1384 5: 'bus',
1385 6: 'train', 1385 6: 'train',
2071 for t in self.getTimeInterval(): 2071 for t in self.getTimeInterval():
2072 if t not in self.userTypes: 2072 if t not in self.userTypes:
2073 self.classifyUserTypeHoGSVMAtInstant(images[t], t, homography, width, height, px, py, minNPixels, rescaleSize, orientations, pixelsPerCell, cellsPerBlock) 2073 self.classifyUserTypeHoGSVMAtInstant(images[t], t, homography, width, height, px, py, minNPixels, rescaleSize, orientations, pixelsPerCell, cellsPerBlock)
2074 # compute P(Speed|Class) 2074 # compute P(Speed|Class)
2075 if speedProbabilities is None or self.aggregatedSpeed < minSpeedEquiprobable: # equiprobable information from speed 2075 if speedProbabilities is None or self.aggregatedSpeed < minSpeedEquiprobable: # equiprobable information from speed
2076 userTypeProbabilities = {userType2Num['car']: 1., userType2Num['pedestrian']: 1., userType2Num['bicycle']: 1.} 2076 userTypeProbabilities = {userType2Num['car']: 1., userType2Num['pedestrian']: 1., userType2Num['cyclist']: 1.}
2077 else: 2077 else:
2078 userTypeProbabilities = {userType2Num[userTypename]: speedProbabilities[userTypename](self.aggregatedSpeed) for userTypename in speedProbabilities} 2078 userTypeProbabilities = {userType2Num[userTypename]: speedProbabilities[userTypename](self.aggregatedSpeed) for userTypename in speedProbabilities}
2079 # compute P(Class|Appearance) 2079 # compute P(Class|Appearance)
2080 nInstantsUserType = {userTypeNum: 0 for userTypeNum in userTypeProbabilities}# number of instants the object is classified as userTypename 2080 nInstantsUserType = {userTypeNum: 0 for userTypeNum in userTypeProbabilities}# number of instants the object is classified as userTypename
2081 nInstantsUserType[userType2Num['unknown']] = 0 2081 nInstantsUserType[userType2Num['unknown']] = 0