Update track_detection.py

scripts/track_detection.py

@@ -79,12 +79,9 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
     out_path = os.path.join(out_loc_dir, seq + ".txt")
 
 
-    if torch.cuda.is_available():
-        device_name = 'cuda:0'
-        device = torch.device(device_name)
-        inference = torch.load(inference_path, map_location=device)
-    else:
-        inference = torch.load(inference_path)
+    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
     with open(json_path, 'r') as f: