Mercurial Hosting > traffic-intelligence
diff scripts/classify-objects.py @ 1243:88eedf79f16a
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)