rachana219 commited on
Commit
2189169
·
1 Parent(s): f0d7ee3

Update track.py

Browse files
Files changed (1) hide show
  1. track.py +1 -1
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