# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1687006102 14400
# Node ID 1f0b1fc172f886f4fd68f899ff1273521ae27ecb
# Parent  5038c357b57faecdcac4f4a2f4f55ddfcdc376fb
adding COCO person and vehicle types

diff -r 5038c357b57f -r 1f0b1fc172f8 trafficintelligence/moving.py
--- a/trafficintelligence/moving.py	Tue May 16 22:12:39 2023 -0400
+++ b/trafficintelligence/moving.py	Sat Jun 17 08:48:22 2023 -0400
@@ -1358,11 +1358,20 @@
                  'car',
                  'pedestrian',
                  'motorcycle',
-                 'bicycle',
+                 'cyclist',
                  'bus',
                  'truck',
                  'automated']
 
+coco2UserTypes = {1: 2, 2: 4, 3: 1, 6: 5, 8: 6}
+cocoUserTypeNames = {1: person,
+                     2: bicycle,
+	             3:	car,
+                     4: motorcycle,
+                     6: bus,
+                     7: train,
+                     8: truck}
+
 userType2Num = utils.inverseEnumeration(userTypeNames)
 
 class CarClassifier: