Spaces:

Suburst
/

DDM_DeepSort

Sleeping

Suburst commited on Nov 22, 2024

Commit

cf6df0c

verified ·

1 Parent(s): aebac23

Update Yolov5_Deepsort/AIDetector_pytorch.py

Files changed (1) hide show

Yolov5_Deepsort/AIDetector_pytorch.py CHANGED Viewed

@@ -16,7 +16,7 @@ class Detector(baseDet):
     def init_model(self):
-        self.weights = 'weights/yolov5s.pt'
         self.device = '0' if torch.cuda.is_available() else 'cpu'
         self.device = select_device(self.device)
         model = attempt_load(self.weights, map_location=self.device)

     def init_model(self):
+        self.weights = 'Yolov5_Deepsort/weights/yolov5s.pt'
         self.device = '0' if torch.cuda.is_available() else 'cpu'
         self.device = select_device(self.device)
         model = attempt_load(self.weights, map_location=self.device)