comparison trafficintelligence/moving.py @ 1228:5654c9173548

merged (bicycle)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 12 Jul 2023 13:21:08 -0400
parents eb3936809ea5 d478d3122804
children d5695e0b59d9
comparison
equal deleted inserted replaced
1227:eb3936809ea5 1228:5654c9173548
1374 'bicycle', 1374 'bicycle',
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',
1386 7: 'truck'} 1386 7: 'truck'}
1387 1387
1388 userType2Num = utils.inverseEnumeration(userTypeNames) 1388 userType2Num = utils.inverseEnumeration(userTypeNames)
1389 1389
1390 class CarClassifier: 1390 class CarClassifier:
1391 def predict(self, hog): 1391 def predict(self, hog):