Suburst commited on
Commit
8f0e6b2
1 Parent(s): 1658a13

Update Yolov5_Deepsort/AIDetector_pytorch.py

Browse files
Yolov5_Deepsort/AIDetector_pytorch.py CHANGED
@@ -21,7 +21,7 @@ class Detector(baseDet):
21
  self.device = select_device(self.device)
22
  model = attempt_load(self.weights, map_location=self.device)
23
  model.to(self.device).eval()
24
- model.half()
25
  # torch.save(model, 'test.pt')
26
  self.m = model
27
  self.names = model.module.names if hasattr(
 
21
  self.device = select_device(self.device)
22
  model = attempt_load(self.weights, map_location=self.device)
23
  model.to(self.device).eval()
24
+ model.float()
25
  # torch.save(model, 'test.pt')
26
  self.m = model
27
  self.names = model.module.names if hasattr(