Spaces:
Running
Running
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.
|
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(
|