Mercurial Hosting > traffic-intelligence
comparison 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 |
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 |