Mercurial Hosting > traffic-intelligence

diff scripts/classify-objects.py @ 1243:88eedf79f16a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
correct classifier.cfg and info classification
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 07 Feb 2024 11:53:00 -0500
parents 4cd8ace3552f
children 00b71da2baac
line wrap: on
line diff
--- a/scripts/classify-objects.py	Wed Feb 07 11:43:03 2024 -0500
+++ b/scripts/classify-objects.py	Wed Feb 07 11:53:00 2024 -0500
@@ -43,6 +43,7 @@
     bikeCarSVM = None
     yolo = YOLO(classifierParams.dlFilename, task='detect')
     useYolo = True
+    print('Using Yolov8 model +'classifierParams.dlFilename)
 else:
     useYolo = False
     pedBikeCarSVM = ml.SVM_load(classifierParams.pedBikeCarSVMFilename)