Mercurial Hosting > traffic-intelligence

diff trafficintelligence/cvutils.py @ 1242:4cd8ace3552f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
major update for classification, allowing the use of neural network classification
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 07 Feb 2024 11:43:03 -0500
parents 3905b393ade0
children 371c718e57d7
line wrap: on
line diff
--- a/trafficintelligence/cvutils.py	Mon Feb 05 17:06:01 2024 -0500
+++ b/trafficintelligence/cvutils.py	Wed Feb 07 11:43:03 2024 -0500
@@ -365,7 +365,8 @@
                                 cv2.rectangle(img, (xCropMin, yCropMin), (xCropMax, yCropMax), cvBlue[colorType], 1)
                             objDescription = '{} '.format(obj.num)
                             if moving.userTypeNames[obj.userType] != 'unknown':
-                                objDescription += moving.userTypeNames[obj.userType][0].upper()
+                                s = moving.userTypeNames[obj.userType]
+                                objDescription += s[0].upper()+s[1]
                             if len(annotations) > 0: # if we loaded annotations, but there is no match
                                 if frameNum not in toMatches[obj.getNum()]:
                                     objDescription += " FA"