Mercurial > hg > nsaunier > traffic-intelligence

comparison 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
comparison
equal deleted inserted replaced
1242:4cd8ace3552f 1243:88eedf79f16a
41 if ultralyticsAvailable and Path(classifierParams.dlFilename).is_file(): # use Yolo 41 if ultralyticsAvailable and Path(classifierParams.dlFilename).is_file(): # use Yolo
42 pedBikeCarSVM = None 42 pedBikeCarSVM = None
43 bikeCarSVM = None 43 bikeCarSVM = None
44 yolo = YOLO(classifierParams.dlFilename, task='detect') 44 yolo = YOLO(classifierParams.dlFilename, task='detect')
45 useYolo = True 45 useYolo = True
46 print('Using Yolov8 model +'classifierParams.dlFilename)
46 else: 47 else:
47 useYolo = False 48 useYolo = False
48 pedBikeCarSVM = ml.SVM_load(classifierParams.pedBikeCarSVMFilename) 49 pedBikeCarSVM = ml.SVM_load(classifierParams.pedBikeCarSVMFilename)
49 bikeCarSVM = ml.SVM_load(classifierParams.bikeCarSVMFilename) 50 bikeCarSVM = ml.SVM_load(classifierParams.bikeCarSVMFilename)
50 51