modify dotrack module

2024-06-03 15:25:39 +08:00
parent d1ea304491
commit f90ef72cbf
19 changed files with 502 additions and 420 deletions
--- a/tracking/trackers/reid/reid_interface.py
+++ b/tracking/trackers/reid/reid_interface.py
@ -46,6 +46,8 @@ class ReIDInterface:
        
        
        self.model = nn.DataParallel(model).to(self.device)
+        
+        self.model = model
        self.model.load_state_dict(torch.load(self.model_path, map_location=self.device))
        self.model.eval()