Spaces:
Runtime error
Runtime error
Commit
·
5707cd3
1
Parent(s):
b2eed0e
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 'yolov8m' in yolo_weights
|
112 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
|
113 |
else:
|
114 |
model = attempt_load(yolo_weights, map_location=device)
|
|
|
108 |
device = select_device(device)
|
109 |
is_seg = '-seg' in str(yolo_weights)
|
110 |
|
111 |
+
if 'yolov8m' in yolo_weights:
|
112 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
|
113 |
else:
|
114 |
model = attempt_load(yolo_weights, map_location=device)
|