Spaces:
Runtime error
Runtime error
Commit
·
2189169
1
Parent(s):
f0d7ee3
Update track.py
Browse files
track.py
CHANGED
@@ -108,7 +108,7 @@ def run(
|
|
108 |
device = select_device(device)
|
109 |
is_seg = '-seg' in str(yolo_weights)
|
110 |
|
111 |
-
if yolo_weights.find('yolov8m') != -1:
|
112 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half) #load yolov8
|
113 |
else:
|
114 |
model = attempt_load(yolo_weights, map_location=device) #load other yolo versions
|
|
|
108 |
device = select_device(device)
|
109 |
is_seg = '-seg' in str(yolo_weights)
|
110 |
|
111 |
+
if str(yolo_weights).find('yolov8m') != -1:
|
112 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half) #load yolov8
|
113 |
else:
|
114 |
model = attempt_load(yolo_weights, map_location=device) #load other yolo versions
|