Update track_detection.py

scripts/track_detection.py

@@ -77,8 +77,8 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
     out_path = os.path.join(out_loc_dir, seq + ".txt")
 
 
-    device_name = '0' if torch.cuda.is_available() else 'cpu'
-    device = select_device(device_name, batch_size=32)
+    device_name = 'cuda:0' if torch.cuda.is_available() else 'cpu'
+    device = torch.device(device_name)
     inference = torch.load(inference_path, map_location=device)
 
     # read detection