Update track.py

track.py

@@ -40,7 +40,6 @@ from ultralytics.yolo.utils.torch_utils import select_device
 from ultralytics.yolo.utils.ops import Profile, non_max_suppression, scale_boxes, process_mask, process_mask_native
 from ultralytics.yolo.utils.plotting import Annotator, colors, save_one_box
 
-from models.experimental import attempt_load
 
 from trackers.multi_tracker_zoo import create_tracker
 
@@ -108,15 +107,9 @@ def run(
     device = select_device(device)
     is_seg = '-seg' in str(yolo_weights)
 
-    if str(yolo_weights).find('yolov8m') != -1:
-        model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half) #load yolov8
-        stride, names, pt = model.stride, model.names, model.pt
-    else:
-        model = attempt_load(yolo_weights, map_location=device) #load other yolo versions
-        stride, names, pt = model.stride, model.names, True
-   
-    #model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
-    #stride, names, pt = model.stride, model.names, model.pt
+  
+    model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
+    stride, names, pt = model.stride, model.names, model.pt
     imgsz = check_imgsz(imgsz, stride=stride)  # check image size
 
     # Dataloader