Spaces:

k20hcmus
/

FishEye8K

Running on Zero

ntsc207 commited on May 12, 2024

Commit

8f4aee7

verified ·

1 Parent(s): dace7ba

Update strong_sort/reid_multibackend.py

Files changed (1) hide show

strong_sort/reid_multibackend.py CHANGED Viewed

@@ -49,7 +49,7 @@ class ReIDDetectMultiBackend(nn.Module):
                 # get rid of dataset information DeepSort model name
                 model_name=model_name,
                 model_path=weights,
-                device=int(device)
             )
             self.extractor.model.half() if fp16 else  self.extractor.model.float()
         elif self.onnx:  # ONNX Runtime

                 # get rid of dataset information DeepSort model name
                 model_name=model_name,
                 model_path=weights,
+                device=str(device)
             )
             self.extractor.model.half() if fp16 else  self.extractor.model.float()
         elif self.onnx:  # ONNX Runtime