Spaces:
Runtime error
Runtime error
Commit
·
eb3543f
1
Parent(s):
3b3152c
Update track_detection.py
Browse files
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:
|