Spaces:

Suburst
/

DDM_DeepSort

Sleeping

Suburst commited on Nov 22, 2024

Commit

8f0e6b2

verified ·

1 Parent(s): 1658a13

Update Yolov5_Deepsort/AIDetector_pytorch.py

Files changed (1) hide show

Yolov5_Deepsort/AIDetector_pytorch.py CHANGED Viewed

@@ -21,7 +21,7 @@ class Detector(baseDet):
         self.device = select_device(self.device)
         model = attempt_load(self.weights, map_location=self.device)
         model.to(self.device).eval()
-        model.half()
         # torch.save(model, 'test.pt')
         self.m = model
         self.names = model.module.names if hasattr(

         self.device = select_device(self.device)
         model = attempt_load(self.weights, map_location=self.device)
         model.to(self.device).eval()
+        model.float()
         # torch.save(model, 'test.pt')
         self.m = model
         self.names = model.module.names if hasattr(