oskarastrom commited on
Commit
eb3543f
·
1 Parent(s): 3b3152c

Update track_detection.py

Browse files
Files changed (1) hide show
  1. scripts/track_detection.py +0 -2
scripts/track_detection.py CHANGED
@@ -82,8 +82,6 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
82
  device_name = 'cuda:0' if torch.cuda.is_available() else 'cpu'
83
  device = torch.device(device_name)
84
  inference = torch.load(inference_path, map_location=device)
85
- print(type(inference))
86
- print(type(inference[0]))
87
 
88
  # read detection
89
  with open(json_path, 'r') as f:
 
82
  device_name = 'cuda:0' if torch.cuda.is_available() else 'cpu'
83
  device = torch.device(device_name)
84
  inference = torch.load(inference_path, map_location=device)
 
 
85
 
86
  # read detection
87
  with open(json_path, 'r') as f: